Searched refs:input_spatial_dims (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 104 auto input_spatial_dims = conv_dnums.input_spatial_dimensions(); in MatchBackwardFilter() local
162 for (int i = 0; i < input_spatial_dims.size(); ++i) { in MatchBackwardFilter()
177 conv->operand(0)->shape().dimensions(input_spatial_dims[i]); in MatchBackwardFilter()
231 for (int i = 0; i < input_spatial_dims.size(); ++i) { in MatchBackwardFilter()
232 backward_conv_dnums.add_input_spatial_dimensions(input_spatial_dims[i]); in MatchBackwardFilter()
346 const auto& input_spatial_dims = dnums.input_spatial_dimensions(); in MatchBackwardInput() local
348 CHECK_EQ(conv->window().dimensions().size(), input_spatial_dims.size()); in MatchBackwardInput()
349 CHECK_EQ(output_spatial_dims.size(), input_spatial_dims.size()); in MatchBackwardInput()
353 for (size_t i = 0; i < input_spatial_dims.size(); ++i) { in MatchBackwardInput()
389 conv->operand(0)->shape().dimensions(input_spatial_dims[i]); in MatchBackwardInput()
[all …]
|
| D | jitrt_custom_calls.h | 48 llvm::ArrayRef<int64_t> input_spatial_dims; member
|
| D | jitrt_custom_calls.cc | 746 for (int64_t d : dims.input_spatial_dims) dns.add_input_spatial_dimensions(d); in GetConvDescriptor()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 90 llvm::SmallVector<int64_t, 4> input_spatial_dims( in ConvertConvDimensionNumbers() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 646 gtl::InlinedVector<DimensionHandle, 2> input_spatial_dims(2); in Conv2DShapeImpl() local
649 absl::MakeSpan(input_spatial_dims), &input_depth_dim, c)); in Conv2DShapeImpl()
738 c, input_spatial_dims[0], filter_rows_dim, dilation_rows, stride_rows, in Conv2DShapeImpl()
741 c, input_spatial_dims[1], filter_cols_dim, dilation_cols, stride_cols, in Conv2DShapeImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1559 std::vector<int64_t> input_spatial_dims; in InferXlaConvOutputShape() local
1562 input_spatial_dims.push_back( in InferXlaConvOutputShape()
1569 RankedTensorType::get(input_spatial_dims, element_type); in InferXlaConvOutputShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 1735 std::vector<int64_t> input_spatial_dims(num_spatial_dims); in InferConvolveShape() local
1737 input_spatial_dims[i] = lhs.dimensions(dnums.input_spatial_dimensions(i)); in InferConvolveShape()
1816 ShapeUtil::MakeShape(lhs.element_type(), input_spatial_dims); in InferConvolveShape()
|