Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops_common.cc135 llvm::Optional<DenseIntElementsAttr> lhs_dilation, in printWindowAttributes()
162 DenseIntElementsAttr &lhs_dilation, in parseWindowAttributes()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_conv_op.cc50 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.cc235 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_t lhs_dilation, int64_t rhs_dilation, in ConvArray3DGeneralDimensionsDilated()
413 std::pair<int64, int64> lhs_dilation, std::pair<int64, int64> rhs_dilation, in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1463 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilated()
1501 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInstruction()
1543 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInputGrad()
1566 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvKernelGrad()
1592 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvForward()
1614 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilatedInternal()
4171 absl::Span<const int64> lhs_dilation, in ConvGeneralDilated()
4187 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvInputGrad()
4201 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in DynamicConvKernelGrad()
4215 absl::Span<const int64> lhs_dilation, in DynamicConvForward()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py250 lhs_dilation, argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc353 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in IsSupportedConvOp() local
491 auto lhs_dilation = conv_op.lhs_dilation().getValue(); in IsSupportedConvOp() local
/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/xla/service/
Dshape_inference.cc2466 absl::Span<const int64> lhs_dilation, in InferWindowFromDimensions()
Dspace_to_batch_converter.cc1022 const int64_t lhs_dilation = win_dims.base_dilation(); in CanPropagate() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc962 mlir::DenseIntElementsAttr lhs_dilation = op.lhs_dilation().getValue(); in EmitConvolutionThunk() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc5071 SmallVector<int64_t, num_spatial_dims> lhs_dilation; in matchAndRewrite() local