Searched refs:output_depth_dim (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| D | fused_conv2d_bias_activation_op.cc | 69 DimensionHandle output_depth_dim = in __anon4e6e467d0102() local
72 int64 output_depth_dim_val = c->Value(output_depth_dim); in __anon4e6e467d0102()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 476 DimensionHandle output_depth_dim = c->Dim( in Conv2DShapeImpl() local
537 output_depth_dim, data_format, c, &output_shape)); in Conv2DShapeImpl()
614 DimensionHandle output_depth_dim = c->Dim(filter_shape, 4); in Conv3DShape() local
636 output_shape = c->MakeShape({batch_size_dim, output_depth_dim, in Conv3DShape()
640 output_cols, output_depth_dim}); in Conv3DShape()
1095 DimensionHandle output_depth_dim = c->Dim(input_shape, 4); in Pool3DShape() local
1113 output_shape = c->MakeShape({batch_size_dim, output_depth_dim, in Pool3DShape()
1117 output_cols, output_depth_dim}); in Pool3DShape()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 416 DimensionHandle output_depth_dim = c->Dim(filter, 3); in CommonFusedConvCalculations() local
431 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in CommonFusedConvCalculations()
922 DimensionHandle output_depth_dim = c->Dim(filter_shape, 2); in __anon3e672dd81d02() local
928 InferenceContext::kUnknownDim, output_depth_dim}); in __anon3e672dd81d02()
934 c->Merge(c->Dim(input_shape, 3), output_depth_dim, &unused)); in __anon3e672dd81d02()
956 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in __anon3e672dd81d02()
|
| D | array_ops.cc | 2471 DimensionHandle output_depth_dim; in __anon7c94107b3902() local
2473 c->Dim(input_shape, 3), ksize_rows * ksize_cols, &output_depth_dim)); in __anon7c94107b3902()
2478 InferenceContext::kUnknownDim, output_depth_dim}); in __anon7c94107b3902()
2497 {batch_size_dim, output_rows, output_cols, output_depth_dim}); in __anon7c94107b3902()
2574 DimensionHandle output_depth_dim; in __anon7c94107b3a02() local
2577 &output_depth_dim)); in __anon7c94107b3a02()
2583 InferenceContext::kUnknownDim, output_depth_dim}); in __anon7c94107b3a02()
2607 output_depth_dim}); in __anon7c94107b3a02()
|