Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_conv_op.cc48 std::vector<int64> lhs_dilation; in Compile() local
54 &lhs_dilation)); in Compile()
79 lhs_dilation, rhs_dilation, dnums_, feature_group_count, in Compile()
Dconv_op_helpers.cc236 std::vector<int64> 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> 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> lhs_dilation(num_spatial_dims, 1); in Compile() local
157 lhs_dilation, rhs_dilation, dims, depth); in Compile()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir48 lhs_dilation = dense<[1,1]> : tensor<2xi64>,
82 lhs_dilation = dense<1> : tensor<2xi64>,
112 lhs_dilation = dense<1> : tensor<2xi64>,
144 lhs_dilation = dense<1> : tensor<2xi64>,
175 lhs_dilation = dense<1> : tensor<2xi64>,
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py247 lhs_dilation, argument
282 lhs_dilation=lhs_dilation,
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h565 absl::Span<const int64> lhs_dilation,
575 absl::Span<const int64> lhs_dilation,
586 absl::Span<const int64> lhs_dilation,
597 absl::Span<const int64> lhs_dilation,
607 absl::Span<const int64> lhs_dilation,
618 absl::Span<const int64> lhs_dilation,
1157 absl::Span<const int64> lhs_dilation,
1167 absl::Span<const int64> lhs_dilation,
1177 absl::Span<const int64> lhs_dilation,
1187 absl::Span<const int64> lhs_dilation,
[all …]
Dxla_builder.cc1451 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilated() argument
1473 lhs_dilation, rhs_dilation)); in ConvGeneralDilated()
1480 padding, lhs_dilation, rhs_dilation, in ConvGeneralDilated()
1489 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInstruction() argument
1505 padding, lhs_dilation, rhs_dilation)); in DynamicConvInstruction()
1531 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInputGrad() argument
1540 lhs, rhs, window_strides, padding, lhs_dilation, rhs_dilation, in DynamicConvInputGrad()
1554 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvKernelGrad() argument
1563 lhs_dilation, rhs_dilation, dimension_numbers, in DynamicConvKernelGrad()
1580 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvForward() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc58 Padding padding, int64 lhs_dilation, int64 rhs_dilation, in ConvArray3DGeneralDimensionsDilated() argument
82 a4dlhs, a4drhs, {kernel_stride, 1}, padding, {lhs_dilation, 1}, in ConvArray3DGeneralDimensionsDilated()
412 std::pair<int64, int64> lhs_dilation, std::pair<int64, int64> rhs_dilation, in ConvArray4DGeneralDimensionsDilated() argument
452 dim.set_base_dilation(lhs_dilation.first); in ConvArray4DGeneralDimensionsDilated()
462 dim2.set_base_dilation(lhs_dilation.second); in ConvArray4DGeneralDimensionsDilated()
Dreference_util.h87 std::pair<int64, int64> lhs_dilation,
101 Padding padding, int64 lhs_dilation, int64 rhs_dilation,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc205 if (conv_op.lhs_dilation().hasValue()) { in IsSupportedConvOp()
206 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in IsSupportedConvOp() local
207 if (!lhs_dilation.isSplat() || lhs_dilation.getSplatValue<int64_t>() != 1) in IsSupportedConvOp()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py757 lhs_dilation = (2, 1)
763 lhs_dilation, rhs_dilation, dimension_numbers)
779 lhs_dilation = (2, 1)
791 lhs_dilation,
810 lhs_dilation = (2, 1)
818 strides, pads, lhs_dilation, rhs_dilation, dimension_numbers)
831 lhs_dilation = (2, 1)
838 lhs_dilation, rhs_dilation, dimension_numbers, feature_group_count)
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h117 absl::Span<const int64> lhs_dilation,
Dmlir_hlo_builder.cc97 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilatedInternal() argument
110 GetI64ElementsAttr(lhs_dilation, &builder_), in ConvGeneralDilatedInternal()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h341 absl::Span<const int64> lhs_dilation,
Dshape_inference.cc2352 absl::Span<const int64> lhs_dilation, in InferWindowFromDimensions() argument
2368 TF_RETURN_IF_ERROR(verify_size(lhs_dilation.size(), "lhs dilation factors")); in InferWindowFromDimensions()
2387 if (!lhs_dilation.empty()) { in InferWindowFromDimensions()
2388 dim->set_base_dilation(lhs_dilation[i]); in InferWindowFromDimensions()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt294 // CHECK-SAME: lhs_dilation = dense<1> : tensor<2xi64>
331 // CHECK-SAME: lhs_dilation = dense<1> : tensor<2xi64>
370 // CHECK-SAME: lhs_dilation = dense<1> : tensor<2xi64>
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py156 lhs_dilation=(1,),
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc202 I64_ELEMENTS_ATTR_TO_VECTOR(lhs_dilation);
722 Convert_padding(op.padding()), Convert_lhs_dilation(op.lhs_dilation()), in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlegalize_hlo.mlir1624 …feature_group_count = 1 : i64, lhs_dilation = dense<1> : tensor<2xi64>, padding = dense<1> : tenso…
1640 …feature_group_count = 207 : i64, lhs_dilation = dense<1> : tensor<2xi64>, padding = dense<1> : ten…
1654 …feature_group_count = 1 : i64, lhs_dilation = dense<1> : tensor<2xi64>, padding = dense<0> : tenso…
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc324 if (op.lhs_dilation() || op.hasWindowReversal()) { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1159 mlir::DenseIntElementsAttr lhs_dilation = op.lhs_dilation().getValue(); in EmitConvolutionThunkFromMlir() local
1173 dim->set_base_dilation(lhs_dilation.getValue<int64>(index)); in EmitConvolutionThunkFromMlir()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base.td1007 OptionalAttr<I64ElementsAttr>:$lhs_dilation,
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4167 SmallVector<int64_t, num_spatial_dims> lhs_dilation; in matchAndRewrite() local
4175 lhs_dilation.push_back(spatial_dim_i.stride); in matchAndRewrite()
4242 /*padding=*/paddings_attr, GetI64ElementsAttr(lhs_dilation, &rewriter), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir3578 // CHECK-NOT: lhs_dilation
3609 // CHECK-NOT: lhs_dilation
3688 // CHECK-SAME: lhs_dilation = dense<1> : tensor<2xi64>,
3751 // CHECK-DAG-SAME: lhs_dilation = dense<1> : tensor<3xi64>
3781 // CHECK-SAME: lhs_dilation = dense<1> : tensor<2xi64>,
3839 // CHECK-DAG-SAME: lhs_dilation = dense<1> : tensor<3xi64>
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dimport.hlotxt237 // CHECK-SAME: lhs_dilation = dense<1> : tensor<2xi64>

12