Home
last modified time | relevance | path

Searched refs:rhs_dilation (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc237 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 …]
Dxla_conv_op.cc49 std::vector<int64> rhs_dilation; in Compile() local
56 &rhs_dilation)); in Compile()
79 lhs_dilation, rhs_dilation, dnums_, feature_group_count, in Compile()
Dextract_image_patches_op.cc129 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/
Dlhlo_gpu_ops.mlir49 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>,
Dhlo-legalize-to-lhlo.mlir540 // CHECK-SAME: rhs_dilation = dense<[1, 2]>
557 rhs_dilation = dense<[1, 2]> : tensor<2xi64>,
Dlhlo-legalize-to-linalg.mlir908 …count = 1 : i64, padding = dense<[[0, 1], [0, 1]]> : tensor<2x2xi64>, rhs_dilation = dense<[1, 2]>…
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py248 rhs_dilation, argument
283 rhs_dilation=rhs_dilation,
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h566 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 …]
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
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()
Dreference_util.h88 std::pair<int64, int64> rhs_dilation, ConvolutionDimensionNumbers dnums);
101 Padding padding, int64 lhs_dilation, int64 rhs_dilation,
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc805 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 …]
Dshape_inference.h342 absl::Span<const int64> rhs_dilation);
Dshape_inference.cc2353 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/
Dlegalize_hlo.cc82 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/
Dxla_client_test.py758 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/
Dmlir_hlo_builder.h118 absl::Span<const int64> rhs_dilation,
Dmlir_hlo_builder.cc97 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/
Dhlo_text_to_lhlo_no_opt.hlotxt297 // 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/
Dlegalize_to_linalg.cc336 if (auto rhs_dilation = op.rhs_dilation()) { in matchAndRewrite() local
337 auto range = rhs_dilation->getAttributeValues(); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py157 rhs_dilation=(2,),
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc203 I64_ELEMENTS_ATTR_TO_VECTOR(rhs_dilation);
723 Convert_rhs_dilation(op.rhs_dilation()), in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4168 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/
Dlegalize_hlo.mlir1624 …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/
Dir_emitter_unnested.cc1160 mlir::DenseIntElementsAttr rhs_dilation = op.rhs_dilation().getValue(); in EmitConvolutionThunkFromMlir() local
1174 dim->set_window_dilation(rhs_dilation.getValue<int64>(index)); in EmitConvolutionThunkFromMlir()

12