Searched refs:output_depth_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 643 DimensionHandle output_depth_dim = c->Dim( in Conv2DShapeImpl() local
674 if (c->ValueKnown(output_depth_dim)) { in Conv2DShapeImpl()
675 int64 output_depth_value = c->Value(output_depth_dim); in Conv2DShapeImpl()
720 output_depth_dim, data_format, c, &output_shape)); in Conv2DShapeImpl()
799 DimensionHandle output_depth_dim = c->Dim(filter_shape, 4); in Conv3DShape() local
813 if (c->ValueKnown(output_depth_dim)) { in Conv3DShape()
814 int64 output_depth_value = c->Value(output_depth_dim); in Conv3DShape()
839 output_shape = c->MakeShape({batch_size_dim, output_depth_dim, in Conv3DShape()
843 output_cols, output_depth_dim}); in Conv3DShape()
1757 DimensionHandle output_depth_dim = c->Dim(input_shape, 4); in Pool3DShape() local
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2630 DimensionHandle output_depth_dim; in __anon42d741193902() local
2632 c->Dim(input_shape, 3), ksize_rows * ksize_cols, &output_depth_dim)); in __anon42d741193902()
2637 InferenceContext::kUnknownDim, output_depth_dim}); in __anon42d741193902()
2656 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in __anon42d741193902()
2733 DimensionHandle output_depth_dim; in __anon42d741193a02() local
2736 &output_depth_dim)); in __anon42d741193a02()
2742 InferenceContext::kUnknownDim, output_depth_dim}); in __anon42d741193a02()
2766 output_depth_dim}); in __anon42d741193a02()
|
| D | nn_ops.cc | 496 DimensionHandle output_depth_dim = c->Dim(filter, 3); in CommonFusedConvCalculations() local
511 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in CommonFusedConvCalculations()
1016 DimensionHandle output_depth_dim = c->Dim(filter_shape, 2); in __anonad10f3361702() local
1022 InferenceContext::kUnknownDim, output_depth_dim}); in __anonad10f3361702()
1028 c->Merge(c->Dim(input_shape, 3), output_depth_dim, &unused)); in __anonad10f3361702()
1050 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in __anonad10f3361702()
|