| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_conv_op.cc | 48 std::vector<int64> lhs_dilation; in Compile() local
54 &lhs_dilation)); in Compile()
79 lhs_dilation, rhs_dilation, dnums_, feature_group_count, in Compile()
|
| D | conv_op_helpers.cc | 287 std::vector<int64> lhs_dilation(attrs.num_spatial_dims, 1); in MakeXlaForwardConvOp() local
319 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
364 std::vector<int64> lhs_dilation(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
376 lhs_dilation[i] = dims.spatial_dims[i].stride; in MakeXlaBackpropInputConvOp()
387 lhs_dilation, rhs_dilation, dnums, in MakeXlaBackpropInputConvOp()
|
| D | extract_image_patches_op.cc | 122 std::vector<int64> lhs_dilation(num_spatial_dims, 1); in Compile() local
151 lhs_dilation, rhs_dilation, dims); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 232 lhs_dilation, argument
267 lhs_dilation=lhs_dilation,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 73 Padding padding, int64 lhs_dilation, int64 rhs_dilation, in ConvArray3DGeneralDimensionsDilated() argument
97 a4dlhs, a4drhs, {kernel_stride, 1}, padding, {lhs_dilation, 1}, in ConvArray3DGeneralDimensionsDilated()
478 std::pair<int64, int64> lhs_dilation, std::pair<int64, int64> rhs_dilation, in ConvArray4DGeneralDimensionsDilated() argument
518 dim.set_base_dilation(lhs_dilation.first); in ConvArray4DGeneralDimensionsDilated()
528 dim2.set_base_dilation(lhs_dilation.second); in ConvArray4DGeneralDimensionsDilated()
|
| D | reference_util.h | 87 std::pair<int64, int64> lhs_dilation,
101 Padding padding, int64 lhs_dilation, int64 rhs_dilation,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 708 lhs_dilation = (2, 1)
711 strides, pads, lhs_dilation, rhs_dilation)
725 lhs_dilation = (2, 1)
729 strides, pads, lhs_dilation, rhs_dilation,
744 lhs_dilation = (2, 1)
750 strides, pads, lhs_dilation, rhs_dilation,
765 lhs_dilation = (2, 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,
1674 def ConvGeneralDilated(self, lhs, rhs, window_strides, padding, lhs_dilation, 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 | 297 absl::Span<const int64> lhs_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()
1214 absl::Span<const int64> lhs_dilation, in MakeWindow() argument
1230 TF_RETURN_IF_ERROR(verify_size(lhs_dilation.size(), "lhs dilation factors")); in MakeWindow()
1249 if (!lhs_dilation.empty()) { in MakeWindow()
1250 dim->set_base_dilation(lhs_dilation[i]); in MakeWindow()
3013 absl::Span<const int64> lhs_dilation, in ConvGeneralDilated() argument
3019 lhs, rhs, window_strides, padding, lhs_dilation, rhs_dilation, in ConvGeneralDilated()
|
| D | xla_builder.h | 412 absl::Span<const int64> lhs_dilation,
651 absl::Span<const int64> lhs_dilation,
807 absl::Span<const int64> lhs_dilation,
1340 absl::Span<const int64> lhs_dilation,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 131 lhs_dilation=(1,),
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 615 | `lhs_dilation` | `ArraySlice<int64>` | n-d lhs dilation factor |
663 The `lhs_dilation` and `rhs_dilation` arguments specify the dilation factor to
|