| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 236 std::vector<int64_t> rhs_dilation(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
341 std::vector<int64_t> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
438 std::vector<int64_t> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
|
| D | extract_image_patches_op.cc | 129 std::vector<int64_t> rhs_dilation(num_spatial_dims); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | replace_cast_hacks_with_tf_xla_ops.cc | 58 Value &rhs_dilation, Value &feature_group_count) { in PrepareXlaConvParams()
130 Value padding, window_strides, lhs_dilation, rhs_dilation, in CreateXLAConvOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 58 Padding padding, int64_t lhs_dilation, int64_t rhs_dilation, in ConvArray3DGeneralDimensionsDilated()
415 std::pair<int64_t, int64_t> rhs_dilation, in ConvArray4DGeneralDimensionsDilated()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1528 absl::Span<const int64_t> rhs_dilation, in ConvGeneralDilated()
1568 absl::Span<const int64_t> rhs_dilation, in DynamicConvInstruction()
1612 absl::Span<const int64_t> rhs_dilation, in DynamicConvInputGrad()
1637 absl::Span<const int64_t> rhs_dilation, in DynamicConvKernelGrad()
1664 absl::Span<const int64_t> rhs_dilation, in DynamicConvForward()
1687 absl::Span<const int64_t> rhs_dilation, in ConvGeneralDilatedInternal()
4312 absl::Span<const int64_t> rhs_dilation, in ConvGeneralDilated()
4329 absl::Span<const int64_t> rhs_dilation, in DynamicConvInputGrad()
4345 absl::Span<const int64_t> rhs_dilation, in DynamicConvKernelGrad()
4360 absl::Span<const int64_t> rhs_dilation, in DynamicConvForward()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1315 llvm::SmallVector<int64_t> rhs_dilation) { in InferWindowFromDimensions()
1600 auto rhs_dilation = op.rhs_dilation(); in PrecheckForXlaConvV2Op() local
1733 auto rhs_dilation = op.rhs_dilation(); in InferShapeForXlaConvV2Op() local
|
| D | legalize_hlo.cc | 415 if (auto rhs_dilation = conv_op.rhs_dilation()) { in matchAndRewrite() local
736 if (auto rhs_dilation = conv_op.rhs_dilation()) { in matchAndRewrite() local
935 auto rhs_dilation = conv_op.rhs_dilation().getValue(); in MatchResizeOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 248 rhs_dilation, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 105 absl::Span<const int64_t> rhs_dilation, in ConvGeneralDilatedInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 1178 const int64_t rhs_dilation = win_dims.window_dilation(); in CanPropagate() local
2998 const int64_t rhs_dilation = in PropagateOnBackpropFilterConv() local
|
| D | shape_inference.cc | 2460 absl::Span<const int64_t> rhs_dilation, in InferWindowFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | jitrt_custom_calls.cc | 693 ArrayRef<int64_t> rhs_dilation; member
|
| D | ir_emitter_unnested.cc | 1020 mlir::DenseIntElementsAttr rhs_dilation = op.getRhsDilation().getValue(); in EmitConvolutionThunk() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4946 SmallVector<int64_t, num_spatial_dims> rhs_dilation; in matchAndRewrite() local
5170 SmallVector<int64_t, num_spatial_dims> rhs_dilation; in matchAndRewrite() local
|