Home
last modified time | relevance | path

Searched defs:rhs_dilation (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_conv_op.cc49 std::vector<int64> rhs_dilation; in Compile() local
Dconv_op_helpers.cc288 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
365 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
445 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
Dextract_image_patches_op.cc123 std::vector<int64> rhs_dilation(num_spatial_dims); in Compile() local
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py233 rhs_dilation, argument
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc73 Padding padding, int64 lhs_dilation, int64 rhs_dilation, in ConvArray3DGeneralDimensionsDilated()
478 std::pair<int64, int64> lhs_dilation, std::pair<int64, int64> rhs_dilation, in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1640 lhs_dilation, rhs_dilation, feature_group_count=1): argument
1675 rhs_dilation, dimension_numbers=None, argument
Dlocal_computation_builder.cc590 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilated()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1169 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilated()
3014 absl::Span<const int64> rhs_dilation, in ConvGeneralDilated()