| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_conv_op.cc | 49 std::vector<int64> rhs_dilation; in Compile() local
|
| D | conv_op_helpers.cc | 237 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
341 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
438 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
|
| D | extract_image_patches_op.cc | 129 std::vector<int64> rhs_dilation(num_spatial_dims); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 58 Padding padding, int64 lhs_dilation, int64 rhs_dilation, in ConvArray3DGeneralDimensionsDilated()
412 std::pair<int64, int64> lhs_dilation, std::pair<int64, int64> rhs_dilation, in ConvArray4DGeneralDimensionsDilated()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1451 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilated()
1489 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInstruction()
1531 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInputGrad()
1554 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvKernelGrad()
1580 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvForward()
1602 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilatedInternal()
4360 absl::Span<const int64> rhs_dilation, in ConvGeneralDilated()
4375 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInputGrad()
4389 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvKernelGrad()
4404 absl::Span<const int64> rhs_dilation, in DynamicConvForward()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 248 rhs_dilation, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 805 const int64 rhs_dilation = win_dims.window_dilation(); in CanPropagate() local
2150 const int64 rhs_dilation = in PropagateOnBackpropFilterConv() local
|
| D | shape_inference.cc | 2353 absl::Span<const int64> rhs_dilation) { in InferWindowFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 97 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilatedInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 82 if (auto rhs_dilation = conv_op.rhs_dilation()) { in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 336 if (auto rhs_dilation = op.rhs_dilation()) { in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4168 SmallVector<int64_t, num_spatial_dims> rhs_dilation; in matchAndRewrite() local
4363 SmallVector<int64_t, num_spatial_dims> rhs_dilation; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1160 mlir::DenseIntElementsAttr rhs_dilation = op.rhs_dilation().getValue(); in EmitConvolutionThunkFromMlir() local
|