Searched refs:in_rows_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 608 DimensionHandle in_rows_dim = c->Dim(input_shape, 2); in Conv3DShape() local
628 c, in_rows_dim, filter_rows_dim, dilation_rows, stride_rows, padding, -1, in Conv3DShape()
680 DimensionHandle in_rows_dim = c->Dim(input_shape, 1); in DepthwiseConv2DNativeShape() local
704 c, in_rows_dim, filter_rows_dim, stride_rows, padding, &output_rows)); in DepthwiseConv2DNativeShape()
761 DimensionHandle in_rows_dim = c->Dim( in AvgPoolShape() local
777 c, in_rows_dim, kernel_rows, stride_rows, padding, &output_rows)); in AvgPoolShape()
913 DimensionHandle in_rows_dim = c->Dim( in MaxPoolShape() local
926 c, in_rows_dim, kernel_rows, stride_rows, padding, &output_rows)); in MaxPoolShape()
1015 DimensionHandle in_rows_dim = c->Dim( in MaxPoolV2Shape() local
1028 c, in_rows_dim, kernel_rows, stride_rows, padding, &output_rows)); in MaxPoolV2Shape()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 412 DimensionHandle in_rows_dim = c->Dim(padded, 1); in CommonFusedConvCalculations() local
426 c, in_rows_dim, filter_rows_dim, stride_rows, padding, &output_rows)); in CommonFusedConvCalculations()
918 DimensionHandle in_rows_dim = c->Dim(input_shape, 1); in __anon3e672dd81d02() local
924 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim) || in __anon3e672dd81d02()
936 auto in_rows = c->Value(in_rows_dim); in __anon3e672dd81d02()
|
| D | array_ops.cc | 2469 DimensionHandle in_rows_dim = c->Dim(input_shape, 1); in __anon7c94107b3902() local
2475 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim)) { in __anon7c94107b3902()
2482 auto in_rows = c->Value(in_rows_dim); in __anon7c94107b3902()
2572 DimensionHandle in_rows_dim = c->Dim(input_shape, 2); in __anon7c94107b3a02() local
2579 if (!c->ValueKnown(in_planes_dim) || !c->ValueKnown(in_rows_dim) || in __anon7c94107b3a02()
2588 auto in_rows = c->Value(in_rows_dim); in __anon7c94107b3a02()
|