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.cc636 DimensionHandle batch_size_dim; in Conv2DShapeImpl() local
640 conv_input_shape, data_format, &batch_size_dim, in Conv2DShapeImpl()
719 ShapeFromDimensions(batch_size_dim, {output_rows, output_cols}, in Conv2DShapeImpl()
789 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in Conv3DShape() local
839 output_shape = c->MakeShape({batch_size_dim, output_depth_dim, in Conv3DShape()
842 output_shape = c->MakeShape({batch_size_dim, output_planes, output_rows, in Conv3DShape()
867 DimensionHandle batch_size_dim; in Conv2DBackpropInputShape() local
871 output_grad_shape, data_format, &batch_size_dim, in Conv2DBackpropInputShape()
899 c->Merge(specified_batch_size_dim, batch_size_dim, &unused)); in Conv2DBackpropInputShape()
915 batch_size_dim, specified_input_grad_spatial_dims, input_grad_depth_dim, in Conv2DBackpropInputShape()
[all …]
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc491 DimensionHandle batch_size_dim = c->Dim(padded, 0); in CommonFusedConvCalculations() local
511 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in CommonFusedConvCalculations()
1011 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon4650176f1702() local
1021 c->MakeShape({batch_size_dim, InferenceContext::kUnknownDim, in __anon4650176f1702()
1050 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in __anon4650176f1702()
Darray_ops.cc2627 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anondb9326b23902() local
2636 c->MakeShape({batch_size_dim, InferenceContext::kUnknownDim, in __anondb9326b23902()
2656 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in __anondb9326b23902()
2729 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anondb9326b23a02() local
2741 c->MakeShape({batch_size_dim, InferenceContext::kUnknownDim, in __anondb9326b23a02()
2765 c->MakeShape({batch_size_dim, output_planes, output_rows, output_cols, in __anondb9326b23a02()