Searched refs:output_batch (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_backprop_filter_test.cc | 35 int64 output_batch, window, window_dilation; member
63 config.output_batch = feature; in GetConv2DTestCases()
85 config.output_batch = feature; in GetConv2DTestCases()
151 spec.window_dilation, spec.window_dilation, spec.output_batch); in BuildHloTextBatchGroupedConvolution2D()
|
| D | convolution_dimension_numbers_test.cc | 39 int64 input_second_spatial, int64 output_batch, int64 output_feature, in CreateConvDimensionNumbers() argument
52 dimension_numbers.set_output_batch_dimension(output_batch); in CreateConvDimensionNumbers()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | pad.h | 69 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local
88 for (int out_b = 0; out_b < output_batch; ++out_b) { in PadImpl()
93 out_b >= output_batch - right_b_padding || in PadImpl()
|
| D | reduce.h | 221 const int output_batch = output_shape.Dims(0); in Mean() local
235 for (int out_b = 0; out_b < output_batch; ++out_b) { in Mean()
265 const int output_batch = output_shape.Dims(0); in Mean() local
291 for (int out_b = 0; out_b < output_batch; ++out_b) { in Mean()
|
| D | reference_ops.h | 88 const int output_batch = output_shape.Dims(0); in DepthToSpace() local
95 TFLITE_DCHECK_EQ(input_batch, output_batch); in DepthToSpace()
97 for (int out_b = 0; out_b < output_batch; ++out_b) { in DepthToSpace()
141 const int output_batch = output_shape.Dims(0); in SpaceToDepth() local
148 TFLITE_DCHECK_EQ(input_batch, output_batch); in SpaceToDepth()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | mean.h | 36 const int output_batch = output_shape.Dims(0); in Mean() local
53 for (int out_b = 0; out_b < output_batch; ++out_b) { in Mean()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | mean.h | 35 const int output_batch = output_shape.Dims(0); in MeanImpl() local
55 for (int out_b = 0; out_b < output_batch; ++out_b) { in MeanImpl()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 930 const int output_batch = output_shape.Dims(0); in MeanImpl() local
951 for (int out_b = 0; out_b < output_batch; ++out_b) { in MeanImpl()
5042 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local
5065 for (int out_b = left_b_padding; out_b < output_batch - right_b_padding; in PadImpl()
5116 Offset(ext_output_shape, output_batch - right_b_padding, 0, 0, 0), in PadImpl()
|