Searched refs:in_rows_dim (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | common_shape_fns.cc | 791 DimensionHandle in_rows_dim = c->Dim(input_shape, 2); in Conv3DShape() local 831 c, in_rows_dim, filter_rows_dim, dilation_rows, stride_rows, padding, -1, in Conv3DShape() 980 DimensionHandle in_rows_dim = c->Dim(input_shape, 1); in DepthwiseConv2DNativeShapeImpl() local 1025 c, in_rows_dim, filter_rows_dim, dilation_rows, stride_rows, padding, in DepthwiseConv2DNativeShapeImpl() 1095 DimensionHandle in_rows_dim = c->Dim( in AvgPoolShape() local 1111 c, in_rows_dim, kernel_rows, stride_rows, padding, &output_rows)); in AvgPoolShape() 1538 DimensionHandle in_rows_dim = c->Dim( in MaxPoolShapeImpl() local 1573 c, in_rows_dim, kernel_rows, /*dilation_rate=*/1, stride_rows, padding, in MaxPoolShapeImpl() 1677 DimensionHandle in_rows_dim = c->Dim( in MaxPoolV2Shape() local 1690 c, in_rows_dim, kernel_rows, stride_rows, padding, &output_rows)); in MaxPoolV2Shape() [all …]
|
/external/tensorflow/tensorflow/core/ops/ |
D | nn_ops.cc | 492 DimensionHandle in_rows_dim = c->Dim(padded, 1); in CommonFusedConvCalculations() local 506 c, in_rows_dim, filter_rows_dim, stride_rows, padding, &output_rows)); in CommonFusedConvCalculations() 1012 DimensionHandle in_rows_dim = c->Dim(input_shape, 1); in __anonad10f3361702() local 1018 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim) || in __anonad10f3361702() 1030 auto in_rows = c->Value(in_rows_dim); in __anonad10f3361702()
|
D | array_ops.cc | 2628 DimensionHandle in_rows_dim = c->Dim(input_shape, 1); in __anon42d741193902() local 2634 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim)) { in __anon42d741193902() 2641 auto in_rows = c->Value(in_rows_dim); in __anon42d741193902() 2731 DimensionHandle in_rows_dim = c->Dim(input_shape, 2); in __anon42d741193a02() local 2738 if (!c->ValueKnown(in_planes_dim) || !c->ValueKnown(in_rows_dim) || in __anon42d741193a02() 2747 auto in_rows = c->Value(in_rows_dim); in __anon42d741193a02()
|