Searched defs:rhs_dilation (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_conv_op.cc | 49 std::vector<int64> rhs_dilation; in Compile() local
|
| D | conv_op_helpers.cc | 257 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
342 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
420 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
|
| D | extract_image_patches_op.cc | 128 std::vector<int64> rhs_dilation(num_spatial_dims); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 58 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/tf2xla/python/ |
| D | xla.py | 241 rhs_dilation, argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1429 rhs_dilation, argument
1481 rhs_dilation, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2505 SmallVector<int64_t, 4> rhs_dilation(num_spatial_dims); in matchAndRewrite() local
2659 SmallVector<int64_t, 4> rhs_dilation(num_spatial_dims); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2261 absl::Span<const int64> rhs_dilation) { in InferWindowFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1281 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilated()
3079 absl::Span<const int64> rhs_dilation, in ConvGeneralDilated()
|