Пример #1
0
 def make_func_for_chunk_size(chunk_size):
     return elementwise.get_put_kernel(
             a_dtype, dest_indices.dtype, vec_count=chunk_size)
Пример #2
0
 def make_func_for_chunk_size(chunk_size):
     return elementwise.get_put_kernel(
             a_dtype, dest_indices.dtype, vec_count=chunk_size)
Пример #3
0
 def make_func_for_chunk_size(chunk_size):
     func = elementwise.get_put_kernel(
             a_dtype, dest_indices.dtype, vec_count=chunk_size)
     func.set_block_shape(*dest_indices._block)
     return func