/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | conv_op_helpers.cc | 237 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local 263 rhs_dilation[i] = attrs.dilations.at(dim); in MakeXlaForwardConvOp() 273 rhs_dilation[i], window_strides[i], attrs.padding, &unused_output_size, in MakeXlaForwardConvOp() 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() 341 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local 364 rhs_dilation[i] = attrs.dilations[dim]; in MakeXlaBackpropInputConvOp() 377 lhs_dilation, rhs_dilation, dnums, in MakeXlaBackpropInputConvOp() 385 padding, lhs_dilation, rhs_dilation, dnums, in MakeXlaBackpropInputConvOp() 438 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local [all …]
|
D | xla_conv_op.cc | 49 std::vector<int64> rhs_dilation; in Compile() local 56 &rhs_dilation)); in Compile() 79 lhs_dilation, rhs_dilation, dnums_, feature_group_count, in Compile()
|
D | extract_image_patches_op.cc | 129 std::vector<int64> rhs_dilation(num_spatial_dims); in Compile() local 145 rhs_dilation[i] = dilations_.at(dim); in Compile() 150 input_shape.dim_size(dim), ksizes_[dim], rhs_dilation[i], in Compile() 157 lhs_dilation, rhs_dilation, dims, depth); in Compile()
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
D | lhlo_gpu_ops.mlir | 49 rhs_dilation = dense<[1,1]> : tensor<2xi64>, 86 rhs_dilation = dense<1> : tensor<2xi64>, 116 rhs_dilation = dense<1> : tensor<2xi64>, 148 rhs_dilation = dense<1> : tensor<2xi64>, 179 rhs_dilation = dense<1> : tensor<2xi64>,
|
D | hlo-legalize-to-lhlo.mlir | 540 // CHECK-SAME: rhs_dilation = dense<[1, 2]> 557 rhs_dilation = dense<[1, 2]> : tensor<2xi64>,
|
D | lhlo-legalize-to-linalg.mlir | 908 …count = 1 : i64, padding = dense<[[0, 1], [0, 1]]> : tensor<2x2xi64>, rhs_dilation = dense<[1, 2]>…
|
/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
D | xla.py | 248 rhs_dilation, argument 283 rhs_dilation=rhs_dilation,
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.h | 566 absl::Span<const int64> rhs_dilation, 576 absl::Span<const int64> rhs_dilation, 587 absl::Span<const int64> rhs_dilation, 598 absl::Span<const int64> rhs_dilation, 608 absl::Span<const int64> rhs_dilation, 619 absl::Span<const int64> rhs_dilation, 1158 absl::Span<const int64> rhs_dilation, 1168 absl::Span<const int64> rhs_dilation, 1178 absl::Span<const int64> rhs_dilation, 1188 absl::Span<const int64> rhs_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 83 {rhs_dilation, 1}, dnums2d); in ConvArray3DGeneralDimensionsDilated() 412 std::pair<int64, int64> lhs_dilation, std::pair<int64, int64> rhs_dilation, in ConvArray4DGeneralDimensionsDilated() argument 451 dim.set_window_dilation(rhs_dilation.first); in ConvArray4DGeneralDimensionsDilated() 461 dim2.set_window_dilation(rhs_dilation.second); in ConvArray4DGeneralDimensionsDilated()
|
D | reference_util.h | 88 std::pair<int64, int64> rhs_dilation, ConvolutionDimensionNumbers dnums); 101 Padding padding, int64 lhs_dilation, int64 rhs_dilation,
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | space_to_batch_converter.cc | 805 const int64 rhs_dilation = win_dims.window_dilation(); in CanPropagate() local 809 if (!last_try || rhs_dilation == 1) { in CanPropagate() 837 if (first_operand->shape().dimensions(new_space_dim) % rhs_dilation != in CanPropagate() 911 rhs_dilation * in CanPropagate() 2150 const int64 rhs_dilation = in PropagateOnBackpropFilterConv() local 2196 const int64 needed_spatial_size = rhs_dilation * new_kernel_split_dim_size; in PropagateOnBackpropFilterConv() 2238 CeilOfRatio(new_split_dim_size, rhs_dilation); in PropagateOnBackpropFilterConv() 2361 rhs_dilation * kernel_new_split_dim_size; in PropagateOnBackpropFilterConv() 2435 old_kernel_split_dim_size * rhs_dilation - (rhs_dilation - 1); in PropagateOnBackpropFilterConv() 2513 ->set_padding_high(-(rhs_dilation - 1)); in PropagateOnBackpropFilterConv() [all …]
|
D | shape_inference.h | 342 absl::Span<const int64> rhs_dilation);
|
D | shape_inference.cc | 2353 absl::Span<const int64> rhs_dilation) { in InferWindowFromDimensions() argument 2369 TF_RETURN_IF_ERROR(verify_size(rhs_dilation.size(), "rhs dilation factors")); in InferWindowFromDimensions() 2392 if (!rhs_dilation.empty()) { in InferWindowFromDimensions() 2393 dim->set_window_dilation(rhs_dilation[i]); in InferWindowFromDimensions()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | legalize_hlo.cc | 82 if (auto rhs_dilation = conv_op.rhs_dilation()) { in matchAndRewrite() local 85 dilation.append(rhs_dilation.getValue().getValues<int64_t>().begin(), in matchAndRewrite() 86 rhs_dilation.getValue().getValues<int64_t>().end()); in matchAndRewrite()
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | xla_client_test.py | 758 rhs_dilation = (1, 1) 763 lhs_dilation, rhs_dilation, dimension_numbers) 780 rhs_dilation = (1, 1) 792 rhs_dilation, 811 rhs_dilation = (1, 1) 818 strides, pads, lhs_dilation, rhs_dilation, dimension_numbers) 832 rhs_dilation = (1, 1) 838 lhs_dilation, rhs_dilation, dimension_numbers, feature_group_count)
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
D | mlir_hlo_builder.h | 118 absl::Span<const int64> rhs_dilation,
|
D | mlir_hlo_builder.cc | 97 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilatedInternal() argument 111 GetI64ElementsAttr(rhs_dilation, &builder_), in ConvGeneralDilatedInternal()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
D | hlo_text_to_lhlo_no_opt.hlotxt | 297 // CHECK_SAME: rhs_dilation = dense<1> : tensor<2xi64> 335 // CHECK-SAME: rhs_dilation = dense<1> : tensor<2xi64> 374 // CHECK-SAME: rhs_dilation = dense<1> : tensor<2xi64>
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
D | legalize_to_linalg.cc | 336 if (auto rhs_dilation = op.rhs_dilation()) { in matchAndRewrite() local 337 auto range = rhs_dilation->getAttributeValues(); in matchAndRewrite()
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | xla_ops_test.py | 157 rhs_dilation=(2,),
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | mlir_hlo_to_hlo.cc | 203 I64_ELEMENTS_ATTR_TO_VECTOR(rhs_dilation); 723 Convert_rhs_dilation(op.rhs_dilation()), in ExportXlaOp()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | legalize_tf.cc | 4168 SmallVector<int64_t, num_spatial_dims> rhs_dilation; in matchAndRewrite() local 4176 rhs_dilation.push_back(dilations[dim]); in matchAndRewrite() 4243 GetI64ElementsAttr(rhs_dilation, &rewriter), in matchAndRewrite() 4363 SmallVector<int64_t, num_spatial_dims> rhs_dilation; in matchAndRewrite() local 4384 rhs_dilation.push_back(spatial_dim_i.stride); in matchAndRewrite() 4448 GetI64ElementsAttr(rhs_dilation, &rewriter), in matchAndRewrite()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
D | legalize_hlo.mlir | 1624 …ense<1> : tensor<2x2xi64>, precision_config = ["DEFAULT", "DEFAULT"], rhs_dilation = dense<1> : te… 1640 …ense<1> : tensor<2x2xi64>, precision_config = ["DEFAULT", "DEFAULT"], rhs_dilation = dense<1> : te… 1654 …ense<0> : tensor<2x2xi64>, precision_config = ["DEFAULT", "DEFAULT"], rhs_dilation = dense<1> : te…
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_unnested.cc | 1160 mlir::DenseIntElementsAttr rhs_dilation = op.rhs_dilation().getValue(); in EmitConvolutionThunkFromMlir() local 1174 dim->set_window_dilation(rhs_dilation.getValue<int64>(index)); in EmitConvolutionThunkFromMlir()
|