Searched refs:kernel_spatial_dims (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 107 auto kernel_spatial_dims = conv_dnums.kernel_spatial_dimensions(); in MatchBackwardFilter() local
148 if ((kernel_spatial_dims.empty() || in MatchBackwardFilter()
149 conv->operand(1)->shape().dimensions(kernel_spatial_dims[0]) <= in MatchBackwardFilter()
236 for (int i = 0; i < kernel_spatial_dims.size(); ++i) { in MatchBackwardFilter()
237 backward_conv_dnums.add_output_spatial_dimensions(kernel_spatial_dims[i]); in MatchBackwardFilter()
|
| D | jitrt_custom_calls.h | 52 llvm::ArrayRef<int64_t> kernel_spatial_dims; member
|
| D | jitrt_custom_calls.cc | 747 for (int64_t d : dims.kernel_spatial_dims) in GetConvDescriptor()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 338 std::vector<int64_t> kernel_spatial_dims(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
360 kernel_spatial_dims[i] = i; in MakeXlaBackpropInputConvOp()
372 filter = xla::Rev(filter, kernel_spatial_dims); in MakeXlaBackpropInputConvOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 93 llvm::SmallVector<int64_t, 4> kernel_spatial_dims( in ConvertConvDimensionNumbers() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5033 SmallVector<int64_t, 4> kernel_spatial_dims; in matchAndRewrite() local
5034 kernel_spatial_dims.resize(num_spatial_dims); in matchAndRewrite()
5035 std::iota(kernel_spatial_dims.begin(), kernel_spatial_dims.end(), 0); in matchAndRewrite()
5040 GetI64ElementsAttr(kernel_spatial_dims, &rewriter)); in matchAndRewrite()
5064 /*kernel_spatial_dimensions=*/kernel_spatial_dims, in matchAndRewrite()
5169 SmallVector<int64_t, num_spatial_dims> kernel_spatial_dims; in matchAndRewrite() local
5181 kernel_spatial_dims.push_back(dim); in matchAndRewrite()
5263 /*input_spatial_dimensions=*/kernel_spatial_dims, in matchAndRewrite()
5270 /*kernel_spatial_dimensions=*/kernel_spatial_dims, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 1743 std::vector<int64_t> kernel_spatial_dims(num_spatial_dims); in InferConvolveShape() local
1745 kernel_spatial_dims[i] = rhs.dimensions(dnums.kernel_spatial_dimensions(i)); in InferConvolveShape()
1805 if (kernel_spatial_dims != window_dims) { in InferConvolveShape()
|