Home
last modified time | relevance | path

Searched defs:output_batch (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmean.h37 const int output_batch = output_shape.Dims(0); in Mean() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dspace_to_depth.h44 const int output_batch = output_shape.Dims(0); in SpaceToDepth() local
Ddepth_to_space.h44 const int output_batch = output_shape.Dims(0); in DepthToSpace() local
Dreduce.h232 const int output_batch = output_shape.Dims(0); in Mean() local
276 const int output_batch = output_shape.Dims(0); in Mean() local
Dpad.h69 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc39 int64 input_second_spatial, int64 output_batch, int64 output_feature, in CreateConvDimensionNumbers()
Dconv_depthwise_backprop_filter_test.cc35 int64 output_batch, window, window_dilation; member
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmean.h35 const int output_batch = output_shape.Dims(0); in MeanImpl() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc993 auto output_batch = padded_shape_splits[0]; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc349 int64_t output_batch = output_shape[0]; in Verify() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h980 const int output_batch = output_shape.Dims(0); in MeanImpl() local
5118 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local