Searched refs:shard_limit (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 364 const int shard_limit = in Compute() local
385 Shard(worker_threads.num_threads, worker_threads.workers, shard_limit, in Compute()
388 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit, in Compute()
390 ConstTensorMap B(out_backprop_data, output_image_size * shard_limit, in Compute()
396 input_data += input_offset * shard_limit; in Compute()
397 out_backprop_data += output_offset * shard_limit; in Compute()
|
| D | conv_grad_ops_3d.cc | 521 const int shard_limit = in Compute() local
563 Shard(worker_threads.num_threads, worker_threads.workers, shard_limit, in Compute()
566 input_backprop_data += input_offset * shard_limit; in Compute()
567 out_backprop_data += output_offset * shard_limit; in Compute()
943 const int shard_limit = in Compute() local
976 Shard(worker_threads.num_threads, worker_threads.workers, shard_limit, in Compute()
979 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit, in Compute()
981 ConstTensorMap B(out_backprop_data, output_image_size * shard_limit, in Compute()
987 input_data += input_offset * shard_limit; in Compute()
988 out_backprop_data += output_offset * shard_limit; in Compute()
|
| D | conv_grad_input_ops.cc | 499 const int shard_limit = in Compute() local
526 Shard(worker_threads.num_threads, worker_threads.workers, shard_limit, in Compute()
529 input_backprop_data += input_offset * shard_limit; in Compute()
530 out_backprop_data += output_offset * shard_limit; in Compute()
|