Searched defs:output_batch (Results 1 – 12 of 12) sorted by relevance
37 const int output_batch = output_shape.Dims(0); in Mean() local
46 const int output_batch = output_shape.Dims(0); in SpaceToDepth() local
44 const int output_batch = output_shape.Dims(0); in DepthToSpace() local
69 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local
285 const int output_batch = output_shape.Dims(0); in Mean() local329 const int output_batch = output_shape.Dims(0); in Mean() local
39 int64_t input_second_spatial, int64_t output_batch, int64_t output_feature, in CreateConvDimensionNumbers()
35 int64 output_batch, window, window_dilation; member
207 typename TTypes<Scalar, 3>::Tensor output_batch( in ComputeAsync() local
35 const int output_batch = output_shape.Dims(0); in MeanImpl() local
1032 auto output_batch = padded_shape_splits[0]; in matchAndRewrite() local
349 int64_t output_batch = output_shape[0]; in Verify() local
978 const int output_batch = output_shape.Dims(0); in MeanImpl() local4655 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local