/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
extract_image_patches_op.cc | 123 std::vector<int64> lhs_dilation(num_spatial_dims, 1); 152 padding, lhs_dilation, rhs_dilation, dims);
|
conv_ops.cc | 260 std::vector<int64> lhs_dilation(num_spatial_dims_, 1); 289 lhs_dilation, rhs_dilation, dims); 411 std::vector<int64> lhs_dilation(num_spatial_dims_); 423 lhs_dilation[i] = dims.spatial_dims[i].stride; 441 lhs_dilation, rhs_dilation, dnums); 639 /*lhs_dilation=*/ones, rhs_dilation, dnums);
|
/external/tensorflow/tensorflow/compiler/xla/ |
reference_util.cc | 103 Padding padding, int64 lhs_dilation, int64 rhs_dilation, 129 a4dlhs, a4drhs, {kernel_stride, 1}, padding, {lhs_dilation, 1}, 510 std::pair<int64, int64> lhs_dilation, std::pair<int64, int64> rhs_dilation, 550 dim.set_base_dilation(lhs_dilation.first); 560 dim2.set_base_dilation(lhs_dilation.second);
|
reference_util.h | 74 std::pair<int64, int64> lhs_dilation, 88 Padding padding, int64 lhs_dilation, int64 rhs_dilation,
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
computation_builder.cc | 133 tensorflow::gtl::ArraySlice<int64> lhs_dilation, 151 !verify_size(lhs_dilation.size(), "lhs dilation factors") || 172 if (!lhs_dilation.empty()) { 173 dim->set_base_dilation(lhs_dilation[i]); 687 tensorflow::gtl::ArraySlice<int64> lhs_dilation, 724 if (!MakeWindow(window_dimensions, window_strides, padding, lhs_dilation, [all...] |
computation_builder.h | 412 tensorflow::gtl::ArraySlice<int64> lhs_dilation, [all...] |
/external/tensorflow/tensorflow/compiler/xla/python/ |
local_computation_builder.cc | 428 tensorflow::gtl::ArraySlice<int64> lhs_dilation, 432 lhs_dilation, rhs_dilation,
|
local_computation_builder.h | 211 tensorflow::gtl::ArraySlice<int64> lhs_dilation,
|
xla_client.py | [all...] |
xla_client_test.py | 502 lhs_dilation = (2, 1) 505 strides, pads, lhs_dilation, rhs_dilation) [all...] |