| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 288 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
304 rhs_dilation[i] = attrs.dilations.at(dim); in MakeXlaForwardConvOp()
314 rhs_dilation[i], window_strides[i], attrs.padding, &unused_output_size, in MakeXlaForwardConvOp()
319 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
365 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
377 rhs_dilation[i] = attrs.dilations[dim]; in MakeXlaBackpropInputConvOp()
387 lhs_dilation, rhs_dilation, dnums, in MakeXlaBackpropInputConvOp()
445 std::vector<int64> rhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
477 rhs_dilation[i] = dims.spatial_dims[i].stride; in MakeXlaBackpropFilterConvOp()
534 rhs_dilation, dnums, in MakeXlaBackpropFilterConvOp()
|
| D | xla_conv_op.cc | 49 std::vector<int64> rhs_dilation; in Compile() local
56 &rhs_dilation)); in Compile()
79 lhs_dilation, rhs_dilation, dnums_, feature_group_count, in Compile()
|
| D | extract_image_patches_op.cc | 123 std::vector<int64> rhs_dilation(num_spatial_dims); in Compile() local
139 rhs_dilation[i] = dilations_.at(dim); in Compile()
144 input_shape.dim_size(dim), ksizes_[dim], rhs_dilation[i], in Compile()
151 lhs_dilation, rhs_dilation, dims); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 233 rhs_dilation, argument
268 rhs_dilation=rhs_dilation,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 73 Padding padding, int64 lhs_dilation, int64 rhs_dilation, in ConvArray3DGeneralDimensionsDilated() argument
98 {rhs_dilation, 1}, dnums2d); in ConvArray3DGeneralDimensionsDilated()
478 std::pair<int64, int64> lhs_dilation, std::pair<int64, int64> rhs_dilation, in ConvArray4DGeneralDimensionsDilated() argument
517 dim.set_window_dilation(rhs_dilation.first); in ConvArray4DGeneralDimensionsDilated()
527 dim2.set_window_dilation(rhs_dilation.second); in ConvArray4DGeneralDimensionsDilated()
|
| D | reference_util.h | 88 std::pair<int64, int64> rhs_dilation, ConvolutionDimensionNumbers dnums);
101 Padding padding, int64 lhs_dilation, int64 rhs_dilation,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 709 rhs_dilation = (1, 1)
711 strides, pads, lhs_dilation, rhs_dilation)
726 rhs_dilation = (1, 1)
729 strides, pads, lhs_dilation, rhs_dilation,
745 rhs_dilation = (1, 1)
750 strides, pads, lhs_dilation, rhs_dilation,
766 rhs_dilation = (1, 1)
770 strides, pads, lhs_dilation, rhs_dilation,
|
| D | xla_client.py | 1640 lhs_dilation, rhs_dilation, feature_group_count=1): argument
1656 lhs, rhs, window_strides, padding, lhs_dilation, rhs_dilation,
1675 rhs_dilation, dimension_numbers=None, argument
1730 lhs, rhs, window_strides, padding, lhs_dilation, rhs_dilation,
|
| D | local_computation_builder.cc | 590 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilated() argument
594 lhs_dilation, rhs_dilation, dimension_numbers, in ConvGeneralDilated()
|
| D | local_computation_builder.h | 298 absl::Span<const int64> rhs_dilation,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1169 absl::Span<const int64> lhs_dilation, absl::Span<const int64> rhs_dilation, in ConvGeneralDilated() argument
1189 lhs_dilation, rhs_dilation)); in ConvGeneralDilated()
1215 absl::Span<const int64> rhs_dilation) const { in MakeWindow()
1231 TF_RETURN_IF_ERROR(verify_size(rhs_dilation.size(), "rhs dilation factors")); in MakeWindow()
1254 if (!rhs_dilation.empty()) { in MakeWindow()
1255 dim->set_window_dilation(rhs_dilation[i]); in MakeWindow()
3014 absl::Span<const int64> rhs_dilation, in ConvGeneralDilated() argument
3019 lhs, rhs, window_strides, padding, lhs_dilation, rhs_dilation, in ConvGeneralDilated()
|
| D | xla_builder.h | 413 absl::Span<const int64> rhs_dilation,
652 absl::Span<const int64> rhs_dilation) const;
808 absl::Span<const int64> rhs_dilation,
1341 absl::Span<const int64> rhs_dilation,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 132 rhs_dilation=(2,),
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 617 | `rhs_dilation` | `ArraySlice<int64>` | n-d rhs dilation factor |
663 The `lhs_dilation` and `rhs_dilation` arguments specify the dilation factor to
|