Searched refs:input_spatial_dims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 183 auto input_spatial_dims = conv_dnums.input_spatial_dimensions(); in MatchBackwardFilter() local
241 for (int i = 0; i < input_spatial_dims.size(); ++i) { in MatchBackwardFilter()
256 conv->operand(0)->shape().dimensions(input_spatial_dims[i]); in MatchBackwardFilter()
310 for (int i = 0; i < input_spatial_dims.size(); ++i) { in MatchBackwardFilter()
311 backward_conv_dnums.add_input_spatial_dimensions(input_spatial_dims[i]); in MatchBackwardFilter()
398 const auto& input_spatial_dims = dnums.input_spatial_dimensions(); in MatchBackwardInput() local
400 CHECK_EQ(conv->window().dimensions().size(), input_spatial_dims.size()); in MatchBackwardInput()
401 CHECK_EQ(output_spatial_dims.size(), input_spatial_dims.size()); in MatchBackwardInput()
405 for (size_t i = 0; i < input_spatial_dims.size(); ++i) { in MatchBackwardInput()
441 conv->operand(0)->shape().dimensions(input_spatial_dims[i]); in MatchBackwardInput()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 106 llvm::SmallVector<int64_t, 4> input_spatial_dims( in ConvertConvDimensionNumbers() local
118 Convert(input_spatial_dims, builder), in ConvertConvDimensionNumbers()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 638 gtl::InlinedVector<DimensionHandle, 2> input_spatial_dims(2); in Conv2DShapeImpl() local
641 absl::MakeSpan(input_spatial_dims), &input_depth_dim, c)); in Conv2DShapeImpl()
711 c, input_spatial_dims[0], filter_rows_dim, dilation_rows, stride_rows, in Conv2DShapeImpl()
714 c, input_spatial_dims[1], filter_cols_dim, dilation_cols, stride_cols, in Conv2DShapeImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 1734 std::vector<int64> input_spatial_dims(num_spatial_dims); in InferConvolveShape() local
1736 input_spatial_dims[i] = lhs.dimensions(dnums.input_spatial_dimensions(i)); in InferConvolveShape()
1814 ShapeUtil::MakeShape(lhs.element_type(), input_spatial_dims); in InferConvolveShape()
|