| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | replace_cast_hacks_with_tf_xla_ops.cc | 57 Value &window_strides, Value &lhs_dilation, in PrepareXlaConvParams()
130 Value padding, window_strides, lhs_dilation, rhs_dilation, in CreateXLAConvOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 128 std::vector<int64_t> lhs_dilation(num_spatial_dims, 1); in Compile() local
|
| D | conv_op_helpers.cc | 235 std::vector<int64_t> lhs_dilation(attrs.num_spatial_dims, 1); in MakeXlaForwardConvOp() local
340 std::vector<int64_t> lhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 58 Padding padding, int64_t lhs_dilation, int64_t rhs_dilation, in ConvArray3DGeneralDimensionsDilated()
414 std::pair<int64_t, int64_t> lhs_dilation, in ConvArray4DGeneralDimensionsDilated()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1527 absl::Span<const int64_t> lhs_dilation, in ConvGeneralDilated()
1567 absl::Span<const int64_t> lhs_dilation, in DynamicConvInstruction()
1611 absl::Span<const int64_t> lhs_dilation, in DynamicConvInputGrad()
1636 absl::Span<const int64_t> lhs_dilation, in DynamicConvKernelGrad()
1663 absl::Span<const int64_t> lhs_dilation, in DynamicConvForward()
1686 absl::Span<const int64_t> lhs_dilation, in ConvGeneralDilatedInternal()
4311 absl::Span<const int64_t> lhs_dilation, in ConvGeneralDilated()
4328 absl::Span<const int64_t> lhs_dilation, in DynamicConvInputGrad()
4344 absl::Span<const int64_t> lhs_dilation, in DynamicConvKernelGrad()
4359 absl::Span<const int64_t> lhs_dilation, in DynamicConvForward()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1314 llvm::SmallVector<int64_t> lhs_dilation, in InferWindowFromDimensions()
1599 auto lhs_dilation = op.lhs_dilation(); in PrecheckForXlaConvV2Op() local
1732 auto lhs_dilation = op.lhs_dilation(); in InferShapeForXlaConvV2Op() local
|
| D | legalize_hlo.cc | 172 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in IsSupportedConvOp() local
832 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in IsSupportedConvOp() local
934 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in MatchResizeOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 247 lhs_dilation, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 104 absl::Span<const int64_t> lhs_dilation, in ConvGeneralDilatedInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2459 absl::Span<const int64_t> lhs_dilation, in InferWindowFromDimensions()
|
| D | space_to_batch_converter.cc | 1179 const int64_t lhs_dilation = win_dims.base_dilation(); in CanPropagate() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | jitrt_custom_calls.cc | 692 ArrayRef<int64_t> lhs_dilation; member
|
| D | ir_emitter_unnested.cc | 1019 mlir::DenseIntElementsAttr lhs_dilation = op.getLhsDilation().getValue(); in EmitConvolutionThunk() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4945 SmallVector<int64_t, num_spatial_dims> lhs_dilation; in matchAndRewrite() local
|