/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | xla_conv_op.cc | 48 std::vector<int64> lhs_dilation; in Compile() local 54 &lhs_dilation)); in Compile() 79 lhs_dilation, rhs_dilation, dnums_, feature_group_count, in Compile()
|
D | conv_op_helpers.cc | 236 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()
|
D | extract_image_patches_op.cc | 128 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/ |
D | lhlo_gpu_ops.mlir | 48 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/ |
D | xla.py | 247 lhs_dilation, argument 282 lhs_dilation=lhs_dilation,
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.h | 565 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 …]
|
D | xla_builder.cc | 1451 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/ |
D | reference_util.cc | 58 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()
|
D | reference_util.h | 87 std::pair<int64, int64> lhs_dilation, 101 Padding padding, int64 lhs_dilation, int64 rhs_dilation,
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | legalize_hlo.cc | 205 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/ |
D | xla_client_test.py | 757 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/ |
D | mlir_hlo_builder.h | 117 absl::Span<const int64> lhs_dilation,
|
D | mlir_hlo_builder.cc | 97 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/ |
D | shape_inference.h | 341 absl::Span<const int64> lhs_dilation,
|
D | shape_inference.cc | 2352 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/ |
D | hlo_text_to_lhlo_no_opt.hlotxt | 294 // 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/ |
D | xla_ops_test.py | 156 lhs_dilation=(1,),
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | mlir_hlo_to_hlo.cc | 202 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/ |
D | legalize_hlo.mlir | 1624 …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/ |
D | legalize_to_linalg.cc | 324 if (op.lhs_dilation() || op.hasWindowReversal()) { in matchAndRewrite()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_unnested.cc | 1159 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/ |
D | hlo_ops_base.td | 1007 OptionalAttr<I64ElementsAttr>:$lhs_dilation,
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | legalize_tf.cc | 4167 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/ |
D | legalize-tf.mlir | 3578 // 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/ |
D | import.hlotxt | 237 // CHECK-SAME: lhs_dilation = dense<1> : tensor<2xi64>
|