Home
last modified time | relevance | path

Searched refs:batch_size_dim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc469 DimensionHandle batch_size_dim; in Conv2DShapeImpl() local
473 conv_input_shape, data_format, &batch_size_dim, in Conv2DShapeImpl()
536 ShapeFromDimensions(batch_size_dim, {output_rows, output_cols}, in Conv2DShapeImpl()
606 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in Conv3DShape() local
636 output_shape = c->MakeShape({batch_size_dim, output_depth_dim, in Conv3DShape()
639 output_shape = c->MakeShape({batch_size_dim, output_planes, output_rows, in Conv3DShape()
679 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in DepthwiseConv2DNativeShape() local
711 c->MakeShape({batch_size_dim, output_depth, output_rows, output_cols}); in DepthwiseConv2DNativeShape()
714 c->MakeShape({batch_size_dim, output_rows, output_cols, output_depth}); in DepthwiseConv2DNativeShape()
759 DimensionHandle batch_size_dim = c->Dim( in AvgPoolShape() local
[all …]
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc411 DimensionHandle batch_size_dim = c->Dim(padded, 0); in CommonFusedConvCalculations() local
431 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in CommonFusedConvCalculations()
917 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon3e672dd81d02() local
927 c->MakeShape({batch_size_dim, InferenceContext::kUnknownDim, in __anon3e672dd81d02()
956 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in __anon3e672dd81d02()
Darray_ops.cc2468 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon7c94107b3902() local
2477 c->MakeShape({batch_size_dim, InferenceContext::kUnknownDim, in __anon7c94107b3902()
2497 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in __anon7c94107b3902()
2570 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon7c94107b3a02() local
2582 c->MakeShape({batch_size_dim, InferenceContext::kUnknownDim, in __anon7c94107b3a02()
2606 c->MakeShape({batch_size_dim, output_planes, output_rows, output_cols, in __anon7c94107b3a02()