Searched defs:output_batch (Results 1 – 6 of 6) sorted by relevance
36 const int output_batch = output_shape.Dims(0); in Mean() local
39 int64 input_second_spatial, int64 output_batch, int64 output_feature, in CreateConvDimensionNumbers()
36 int64 output_batch, window, window_dilation; member
310 const int output_batch = output_shape.Dims(0); in DepthToSpace() local363 const int output_batch = output_shape.Dims(0); in SpaceToDepth() local3216 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local3619 const int output_batch = output_shape.Dims(0); in Mean() local3663 const int output_batch = output_shape.Dims(0); in Mean() local
1766 const int output_batch = output_shape.Dims(0); in MeanImpl() local5897 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local
6222 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local