Home
last modified time | relevance | path

Searched defs:lhs_dilation (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_conv_op.cc48 std::vector<int64> lhs_dilation; in Compile() local
Dextract_image_patches_op.cc128 std::vector<int64> lhs_dilation(num_spatial_dims, 1); in Compile() local
Dconv_op_helpers.cc236 std::vector<int64> lhs_dilation(attrs.num_spatial_dims, 1); in MakeXlaForwardConvOp() local
340 std::vector<int64> lhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc58 Padding padding, int64 lhs_dilation, int64 rhs_dilation, in ConvArray3DGeneralDimensionsDilated()
412 std::pair<int64, int64> lhs_dilation, std::pair<int64, int64> rhs_dilation, in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1451 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilated()
1489 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInstruction()
1531 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInputGrad()
1554 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvKernelGrad()
1580 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvForward()
1602 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilatedInternal()
4359 absl::Span<const int64> lhs_dilation, in ConvGeneralDilated()
4375 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInputGrad()
4389 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvKernelGrad()
4403 absl::Span<const int64> lhs_dilation, in DynamicConvForward()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py247 lhs_dilation, argument
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc97 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilatedInternal()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc206 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in IsSupportedConvOp() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc2352 absl::Span<const int64> lhs_dilation, in InferWindowFromDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1159 mlir::DenseIntElementsAttr lhs_dilation = op.lhs_dilation().getValue(); in EmitConvolutionThunkFromMlir() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4167 SmallVector<int64_t, num_spatial_dims> lhs_dilation; in matchAndRewrite() local