Searched defs:in_batch (Results 1 – 10 of 10) 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
145 const int64 in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute() local
|
| D | conv_ops_3d.cc | 140 const int64 in_batch = GetTensorDim(input, data_format_, 'N'); in Compute() local
222 const int64 in_batch = GetTensorDim(input, data_format, 'N'); in launch() local
|
| D | fused_batch_norm_op.cc | 122 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
260 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
352 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
1271 const int64 in_batch = GetTensorDim(x, tensor_format_, 'N'); in ComputeWithReservedSpace() local
1432 const int64 in_batch = GetTensorDim(x, tensor_format_, 'N'); in ComputeWithReservedSpace() 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 | 167 const int64 in_batch = GetTensorDim(tensor_in, data_format_, 'N'); in Compute() local
|
| D | conv_ops.cc | 666 const int64 in_batch = GetTensorDim(input, data_format, 'N'); in operator ()() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_to_space_nd.h | 73 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND() local
|
| /external/kernel-headers/original/uapi/linux/ |
| D | bpf.h | 525 __aligned_u64 in_batch; /* start batch, member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 4038 int in_batch = 10; in TEST_F() member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5028 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND() local
|