Home
last modified time | relevance | path

Searched refs:lhs_dilation (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dreplace_cast_hacks_with_tf_xla_ops.cc57 Value &window_strides, Value &lhs_dilation, in PrepareXlaConvParams() argument
66 lhs_dilation = Create1DConstValue<int32_t>(builder, loc, {1, 1}); in PrepareXlaConvParams()
130 Value padding, window_strides, lhs_dilation, rhs_dilation, in CreateXLAConvOp() local
134 /*lhs_dilation=*/lhs_dilation, in CreateXLAConvOp()
146 /*rhs=*/filter, window_strides, padding, lhs_dilation, in CreateXLAConvOp()
Dtf_quant_ops.td143 TF_Int32Tensor:$lhs_dilation,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc235 std::vector<int64_t> lhs_dilation(attrs.num_spatial_dims, 1); in MakeXlaForwardConvOp() local
279 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
287 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
340 std::vector<int64_t> lhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
363 lhs_dilation[i] = dims.spatial_dims[i].stride; in MakeXlaBackpropInputConvOp()
377 lhs_dilation, rhs_dilation, dnums, in MakeXlaBackpropInputConvOp()
385 padding, lhs_dilation, rhs_dilation, dnums, in MakeXlaBackpropInputConvOp()
Dextract_image_patches_op.cc128 std::vector<int64_t> lhs_dilation(num_spatial_dims, 1); in Compile() local
157 lhs_dilation, rhs_dilation, dims, depth); in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py247 lhs_dilation, argument
294 lhs_dilation=lhs_dilation,
307 lhs_dilation=lhs_dilation,
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/
Dverifier_conv_op.mlir23 lhs_dilation = dense<1> : tensor<2xi64>,
50 lhs_dilation = dense<1> : tensor<2xi64>,
208 lhs_dilation = dense<1> : tensor<2xi64>,
236 lhs_dilation = dense<1> : tensor<2xi64>,
264 lhs_dilation = dense<1> : tensor<2xi64>,
523 lhs_dilation = dense<1> : tensor<2xi64>,
551 lhs_dilation = dense<1> : tensor<2xi64>,
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/tests/
Dverify_conv.mlir23 lhs_dilation = dense<1> : tensor<2xi64>,
50 lhs_dilation = dense<1> : tensor<2xi64>,
208 lhs_dilation = dense<1> : tensor<2xi64>,
236 lhs_dilation = dense<1> : tensor<2xi64>,
264 lhs_dilation = dense<1> : tensor<2xi64>,
523 lhs_dilation = dense<1> : tensor<2xi64>,
551 lhs_dilation = dense<1> : tensor<2xi64>,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc171 if (conv_op.lhs_dilation().has_value()) { in IsSupportedConvOp()
172 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in IsSupportedConvOp() local
173 if (!lhs_dilation.isSplat() || lhs_dilation.getSplatValue<int64_t>() != 1) in IsSupportedConvOp()
299 conv_op.lhs_dilation().getValue().getValues<int64_t>()) { in matchAndRewrite()
730 conv_op.lhs_dilation().getValue().getValues<int64_t>().begin(), in matchAndRewrite()
731 conv_op.lhs_dilation().getValue().getValues<int64_t>().end()); in matchAndRewrite()
828 if (!conv_op.lhs_dilation().has_value()) { in IsSupportedConvOp()
832 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in IsSupportedConvOp() local
833 if (lhs_dilation.isSplat() && lhs_dilation.getSplatValue<int64_t>() == 1) in IsSupportedConvOp()
934 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in MatchResizeOp() local
[all …]
Dshape_inference.cc1314 llvm::SmallVector<int64_t> lhs_dilation, in InferWindowFromDimensions() argument
1332 verify_size(lhs_dilation.size(), "lhs dilation factors") && in InferWindowFromDimensions()
1352 if (!lhs_dilation.empty()) { in InferWindowFromDimensions()
1353 dim->set_base_dilation(lhs_dilation[i]); in InferWindowFromDimensions()
1599 auto lhs_dilation = op.lhs_dilation(); in PrecheckForXlaConvV2Op() local
1609 lhs_dilation.getType().cast<TensorType>().hasStaticShape() && in PrecheckForXlaConvV2Op()
1732 auto lhs_dilation = op.lhs_dilation(); in InferShapeForXlaConvV2Op() local
1740 matchPattern(lhs_dilation, m_Constant(&lhs_dilation_attr)) && in InferShapeForXlaConvV2Op()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h566 absl::Span<const int64_t> lhs_dilation,
577 absl::Span<const int64_t> lhs_dilation,
588 absl::Span<const int64_t> lhs_dilation,
599 absl::Span<const int64_t> lhs_dilation,
609 absl::Span<const int64_t> lhs_dilation,
620 absl::Span<const int64_t> lhs_dilation,
1199 absl::Span<const int64_t> lhs_dilation,
1209 absl::Span<const int64_t> lhs_dilation,
1219 absl::Span<const int64_t> lhs_dilation,
1229 absl::Span<const int64_t> lhs_dilation,
[all …]
Dxla_builder.cc1527 absl::Span<const int64_t> lhs_dilation, in ConvGeneralDilated() argument
1551 lhs_dilation, rhs_dilation, window_reversal)); in ConvGeneralDilated()
1558 padding, lhs_dilation, rhs_dilation, in ConvGeneralDilated()
1567 absl::Span<const int64_t> lhs_dilation, in DynamicConvInstruction() argument
1585 padding, lhs_dilation, rhs_dilation)); in DynamicConvInstruction()
1611 absl::Span<const int64_t> lhs_dilation, in DynamicConvInputGrad() argument
1621 lhs, rhs, window_strides, padding, lhs_dilation, rhs_dilation, in DynamicConvInputGrad()
1636 absl::Span<const int64_t> lhs_dilation, in DynamicConvKernelGrad() argument
1646 lhs_dilation, rhs_dilation, dimension_numbers, in DynamicConvKernelGrad()
1663 absl::Span<const int64_t> lhs_dilation, in DynamicConvForward() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc58 Padding padding, int64_t lhs_dilation, int64_t rhs_dilation, in ConvArray3DGeneralDimensionsDilated() argument
82 a4dlhs, a4drhs, {kernel_stride, 1}, padding, {lhs_dilation, 1}, in ConvArray3DGeneralDimensionsDilated()
414 std::pair<int64_t, int64_t> lhs_dilation, in ConvArray4DGeneralDimensionsDilated() argument
455 dim.set_base_dilation(lhs_dilation.first); in ConvArray4DGeneralDimensionsDilated()
465 dim2.set_base_dilation(lhs_dilation.second); in ConvArray4DGeneralDimensionsDilated()
Dreference_util.h84 std::pair<int64_t, int64_t> lhs_dilation,
99 int64_t kernel_stride, Padding padding, int64_t lhs_dilation,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dshape_inference.mlir1714 %lhs_dilation = "tf.Const"() {value = dense<[4, 1, 1]> : tensor<3xi32>} : () -> tensor<3xi32>
1718 …%0 = "tf.XlaConvV2"(%lhs, %rhs, %strides, %padding, %lhs_dilation, %rhs_dilation, %feature_group_c…
1726 %lhs_dilation = "tf.Const"() {value = dense<[4, 1, 1]> : tensor<3xi32>} : () -> tensor<3xi32>
1730 …%0 = "tf.XlaConvV2"(%lhs, %rhs, %strides, %padding, %lhs_dilation, %rhs_dilation, %feature_group_c…
1738 %lhs_dilation = "tf.Const"() {value = dense<[4, 1, 1]> : tensor<3xi32>} : () -> tensor<3xi32>
1742 …%0 = "tf.XlaConvV2"(%lhs, %rhs, %strides, %padding, %lhs_dilation, %rhs_dilation, %feature_group_c…
1750 %lhs_dilation = "tf.Const"() {value = dense<[4, 1, 1]> : tensor<3xi32>} : () -> tensor<3xi32>
1754 …%0 = "tf.XlaConvV2"(%lhs, %rhs, %strides, %padding, %lhs_dilation, %rhs_dilation, %feature_group_c…
1762 %lhs_dilation = "tf.Const"() {value = dense<[4, 1, 1]> : tensor<3xi32>} : () -> tensor<3xi32>
1766 …%0 = "tf.XlaConvV2"(%lhs, %rhs, %strides, %padding, %lhs_dilation, %rhs_dilation, %feature_group_c…
[all …]
Dlegalize_hlo.mlir1645 lhs_dilation = dense<1> : tensor<1xi64>,
1677 lhs_dilation = dense<1> : tensor<1xi64>,
1697 lhs_dilation = dense<1> : tensor<1xi64>,
1717 lhs_dilation = dense<1> : tensor<1xi64>,
1737 lhs_dilation = dense<1> : tensor<1xi64>,
1763 …>, feature_group_count = 1 : i64, lhs_dilation = dense<1> : tensor<2xi64>, padding = dense<1> : te…
1792 …>, feature_group_count = 1 : i64, lhs_dilation = dense<1> : tensor<2xi64>, padding = dense<1> : te…
1815 …>, feature_group_count = 1 : i64, lhs_dilation = dense<1> : tensor<2xi64>, padding = dense<1> : te…
1845 …>, feature_group_count = 1 : i64, lhs_dilation = dense<1> : tensor<2xi64>, padding = dense<[[4, -2…
1871 …>, feature_group_count = 207 : i64, lhs_dilation = dense<1> : tensor<2xi64>, padding = dense<1> : …
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py968 lhs_dilation = (2, 1)
974 lhs_dilation, rhs_dilation, dimension_numbers)
990 lhs_dilation = (2, 1)
1002 lhs_dilation,
1021 lhs_dilation = (2, 1)
1029 strides, pads, lhs_dilation, rhs_dilation, dimension_numbers)
1042 lhs_dilation = (2, 1)
1049 lhs_dilation, rhs_dilation, dimension_numbers, feature_group_count)
1070 lhs_dilation = (2, 1)
1080 lhs_dilation,
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo_gpu/IR/
Dlhlo_gpu_ops.td57 $lhs_dilation, $rhs_dilation,
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h372 absl::Span<const int64_t> lhs_dilation,
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h124 absl::Span<const int64_t> lhs_dilation,
Dmlir_hlo_builder.cc104 absl::Span<const int64_t> lhs_dilation, in ConvGeneralDilatedInternal() argument
118 GetI64ElementsAttr(lhs_dilation, &builder_), in ConvGeneralDilatedInternal()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/lhlo_gpu/
Dlhlo_gpu_ops.mlir27 lhs_dilation = dense<[1,1]> : tensor<2xi64>,
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DStablehloAttrs.td192 OptionalAttr<I64ElementsAttr>:$lhs_dilation,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Djitrt_custom_calls.cc692 ArrayRef<int64_t> lhs_dilation; member
765 dim->set_base_dilation(w.lhs_dilation[index]); in GetConvDescriptor()
814 ArrayRef<int64_t> lhs_dilation, ArrayRef<int64_t> rhs_dilation, in operator ()()
833 {window_strides, padding, lhs_dilation, rhs_dilation, window_reversal}, in operator ()()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base.td226 OptionalAttr<I64ElementsAttr>:$lhs_dilation,
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/lhlo_to_jitrt/
Dconv_to_jitrt.mlir38 // CHECK-DAG: lhs_dilation = dense<1> : tensor<2xi64>

12