| /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
74 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND()
75 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND()
91 input1_data + Offset(input1_shape, in_batch, in_h, in_w, 0); in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 44 const int64 in_batch = GetTensorDim(tensor_in, data_format, 'N'); in Compute() local
74 input_desc.set_count(in_batch) in Compute()
78 output_desc.set_count(in_batch) in Compute()
145 const int64 in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute() local
208 orig_output_desc.set_count(in_batch) in Compute()
213 orig_input_desc.set_count(in_batch) in Compute()
|
| D | conv_ops_3d.cc | 140 const int64 in_batch = GetTensorDim(input, data_format_, 'N'); in Compute() local
170 data_format_, in_batch, {{out[0], out[1], out[2]}}, out_depth); in Compute()
222 const int64 in_batch = GetTensorDim(input, data_format, 'N'); in launch() local
256 const uint64 m = in_batch * in_planes * in_rows * in_cols; in launch()
283 const uint64 m = in_batch; in launch()
323 data_format, in_batch, {{new_in_planes, new_in_rows, new_in_cols}}, in launch()
358 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth); in launch()
398 input_desc.set_count(in_batch) in launch()
405 output_desc.set_count(in_batch) in launch()
460 ShapeFromFormat(FORMAT_NCHW, in_batch, in launch()
[all …]
|
| D | fused_batch_norm_op.cc | 122 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
128 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
133 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
260 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
266 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
271 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
352 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
358 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
363 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
368 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
[all …]
|
| D | conv_ops_fused_impl.h | 460 const int64 in_batch = GetTensorDim(input, params.data_format, 'N');
508 ShapeFromFormat(params.data_format, in_batch,
543 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths);
573 input_desc.set_count(in_batch)
646 {in_batch, // batch
|
| D | conv_ops.cc | 666 const int64 in_batch = GetTensorDim(input, data_format, 'N'); in operator ()() local
683 const uint64 m = in_batch * in_rows * in_cols; in operator ()()
711 const uint64 m = in_batch; in operator ()()
808 ShapeFromFormat(data_format, in_batch, new_in_rows, in operator ()()
841 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths); in operator ()()
880 input_desc.set_count(in_batch) in operator ()()
967 ConvParameters conv_parameters = {in_batch, // batch in operator ()()
|
| D | fractional_avg_pool_op.cc | 253 const int64 in_batch = orig_input_tensor_shape_flat(0); in Compute() local
274 in_cols * in_rows * in_batch); in Compute()
|
| D | pooling_ops_3d.cc | 167 const int64 in_batch = GetTensorDim(tensor_in, data_format_, 'N'); in Compute() local
185 TensorShape out_shape = ShapeFromFormat(data_format_, in_batch, in Compute()
|
| /external/kernel-headers/original/uapi/linux/ |
| D | bpf.h | 525 __aligned_u64 in_batch; /* start batch, 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
5029 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND()
5030 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND()
5059 input1_data + Offset(input1_shape, in_batch, in_h, in_w, 0); in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 4038 int in_batch = 10; in TEST_F() member
4099 in_dims.push_back(options.in_batch); in TEST_F()
|