Exemplo n.º 1
0
 def make_func_for_chunk_size(chunk_size):
     return elementwise.get_put_kernel(
             a_dtype, dest_indices.dtype, vec_count=chunk_size)
Exemplo n.º 2
0
 def make_func_for_chunk_size(chunk_size):
     return elementwise.get_put_kernel(
             a_dtype, dest_indices.dtype, vec_count=chunk_size)
Exemplo n.º 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