Searched defs:in_batch (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 44 const int64 in_batch = GetTensorDim(tensor_in, data_format, 'N'); in Compute() local
130 const int64 in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute() local
|
| D | conv_ops_3d.cc | 134 const int64 in_batch = GetTensorDim(input, data_format_, 'N'); in Compute() local
213 const int64 in_batch = GetTensorDim(input, data_format, 'N'); in launch() local
|
| D | fractional_avg_pool_op.cc | 253 const int64 in_batch = orig_input_tensor_shape_flat(0); in Compute() local
|
| D | pooling_ops_3d.cc | 172 const int64 in_batch = GetTensorDim(tensor_in, data_format_, 'N'); in Compute() local
|
| D | conv_ops.cc | 598 const int64 in_batch = GetTensorDim(input, data_format, 'N'); in operator ()() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 3274 int in_batch = 10; in TEST_F() member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3161 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5807 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6135 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND() local
|