Home
last modified time | relevance | path

Searched refs:output_depth_dim (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/ops/
Dfused_conv2d_bias_activation_op.cc69 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/
Dcommon_shape_fns.cc476 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/
Dnn_ops.cc416 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()
Darray_ops.cc2471 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()