Searched refs:input_spatial_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_rewriter.cc | 120 auto input_spatial_dims = conv_dnums.input_spatial_dimensions(); in MatchBackwardFilter() local
163 for (int i = 0; i < input_spatial_dims.size(); ++i) { in MatchBackwardFilter()
178 conv->operand(0)->shape().dimensions(input_spatial_dims[i]); in MatchBackwardFilter()
232 for (int i = 0; i < input_spatial_dims.size(); ++i) { in MatchBackwardFilter()
233 backward_conv_dnums.add_input_spatial_dimensions(input_spatial_dims[i]); in MatchBackwardFilter()
327 const auto& input_spatial_dims = dnums.input_spatial_dimensions(); in MatchBackwardInput() local
329 CHECK_EQ(conv->window().dimensions().size(), input_spatial_dims.size()); in MatchBackwardInput()
330 CHECK_EQ(output_spatial_dims.size(), input_spatial_dims.size()); in MatchBackwardInput()
334 for (size_t i = 0; i < input_spatial_dims.size(); ++i) { in MatchBackwardInput()
371 conv->operand(0)->shape().dimensions(input_spatial_dims[i]); in MatchBackwardInput()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 471 gtl::InlinedVector<DimensionHandle, 2> input_spatial_dims(2); in Conv2DShapeImpl() local
474 absl::MakeSpan(input_spatial_dims), &input_depth_dim, c)); in Conv2DShapeImpl()
528 c, input_spatial_dims[0], filter_rows_dim, dilation_rows, stride_rows, in Conv2DShapeImpl()
531 c, input_spatial_dims[1], filter_cols_dim, dilation_cols, stride_cols, in Conv2DShapeImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 1694 std::vector<int64> input_spatial_dims(num_spatial_dims); in InferConvolveShape() local
1696 input_spatial_dims[i] = lhs.dimensions(dnums.input_spatial_dimensions(i)); in InferConvolveShape()
1776 ShapeUtil::MakeShape(lhs.element_type(), input_spatial_dims); in InferConvolveShape()
|