Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc609 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in Conv3DShape() local
631 c, in_cols_dim, filter_cols_dim, dilation_cols, stride_cols, padding, -1, in Conv3DShape()
681 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in DepthwiseConv2DNativeShape() local
706 c, in_cols_dim, filter_cols_dim, stride_cols, padding, &output_cols)); in DepthwiseConv2DNativeShape()
763 DimensionHandle in_cols_dim = c->Dim( in AvgPoolShape() local
779 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols)); in AvgPoolShape()
915 DimensionHandle in_cols_dim = c->Dim( in MaxPoolShape() local
928 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols)); in MaxPoolShape()
1017 DimensionHandle in_cols_dim = c->Dim( in MaxPoolV2Shape() local
1030 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols)); in MaxPoolV2Shape()
[all …]
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc413 DimensionHandle in_cols_dim = c->Dim(padded, 2); in CommonFusedConvCalculations() local
428 c, in_cols_dim, filter_cols_dim, stride_cols, padding, &output_cols)); in CommonFusedConvCalculations()
919 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anon3e672dd81d02() local
924 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim) || in __anon3e672dd81d02()
937 auto in_cols = c->Value(in_cols_dim); in __anon3e672dd81d02()
Darray_ops.cc2470 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anon7c94107b3902() local
2475 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim)) { in __anon7c94107b3902()
2483 auto in_cols = c->Value(in_cols_dim); in __anon7c94107b3902()
2573 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in __anon7c94107b3a02() local
2580 !c->ValueKnown(in_cols_dim)) { in __anon7c94107b3a02()
2589 auto in_cols = c->Value(in_cols_dim); in __anon7c94107b3a02()