Searched refs:in_cols_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 792 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in Conv3DShape() local
834 c, in_cols_dim, filter_cols_dim, dilation_cols, stride_cols, padding, -1, in Conv3DShape()
981 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in DepthwiseConv2DNativeShapeImpl() local
1028 c, in_cols_dim, filter_cols_dim, dilation_cols, stride_cols, padding, in DepthwiseConv2DNativeShapeImpl()
1097 DimensionHandle in_cols_dim = c->Dim( in AvgPoolShape() local
1113 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols)); in AvgPoolShape()
1540 DimensionHandle in_cols_dim = c->Dim( in MaxPoolShapeImpl() local
1576 c, in_cols_dim, kernel_cols, /*dilation_rate=*/1, stride_cols, padding, in MaxPoolShapeImpl()
1679 DimensionHandle in_cols_dim = c->Dim( in MaxPoolV2Shape() local
1692 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols)); in MaxPoolV2Shape()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 493 DimensionHandle in_cols_dim = c->Dim(padded, 2); in CommonFusedConvCalculations() local
508 c, in_cols_dim, filter_cols_dim, stride_cols, padding, &output_cols)); in CommonFusedConvCalculations()
1013 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anonad10f3361702() local
1018 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim) || in __anonad10f3361702()
1031 auto in_cols = c->Value(in_cols_dim); in __anonad10f3361702()
|
| D | array_ops.cc | 2629 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anon42d741193902() local
2634 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim)) { in __anon42d741193902()
2642 auto in_cols = c->Value(in_cols_dim); in __anon42d741193902()
2732 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in __anon42d741193a02() local
2739 !c->ValueKnown(in_cols_dim)) { in __anon42d741193a02()
2748 auto in_cols = c->Value(in_cols_dim); in __anon42d741193a02()
|