Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:window_dimensions (Results 1 - 10 of 10) sorted by null

/external/tensorflow/tensorflow/compiler/xla/client/
padding.cc	28 tensorflow::gtl::ArraySlice<int64> window_dimensions, 30 bool ok = input_dimensions.size() == window_dimensions.size() && 36 input_dimensions.size(), window_dimensions.size(), 44 tensorflow::gtl::ArraySlice<int64> window_dimensions, 46 TF_CHECK_OK(ValidatePaddingValues(input_dimensions, window_dimensions, 51 low_high_padding.resize(window_dimensions.size(), {0, 0}); 57 int64 window_dimension = window_dimensions[i];
padding.h	46 tensorflow::gtl::ArraySlice<int64> window_dimensions, 58 // window_dimensions, and strides must match, which is equal to the number 62 tensorflow::gtl::ArraySlice<int64> window_dimensions,
computation_builder.cc	130 tensorflow::gtl::ArraySlice<int64> window_dimensions, 136 if (x == 0 || x == window_dimensions.size()) { 143 window_dimensions.size(), "\nNumber of ", x_name, ": ", x, 157 for (size_t i = 0; i < window_dimensions.size(); i++) { 159 dim->set_size(window_dimensions[i]); 661 std::vector<int64> window_dimensions( 663 for (std::vector<int64>::size_type i = 0; i < window_dimensions.size(); ++i) { 664 window_dimensions[i] = 669 MakePadding(base_area_dimensions, window_dimensions, 711 std::vector<int64> window_dimensions(     [all...]
computation_builder.h	545 tensorflow::gtl::ArraySlice<int64> window_dimensions, 553 tensorflow::gtl::ArraySlice<int64> window_dimensions, 565 tensorflow::gtl::ArraySlice<int64> window_dimensions, 574 tensorflow::gtl::ArraySlice<int64> window_dimensions,     [all...]
/external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc	393 tensorflow::gtl::ArraySlice<int64> window_dimensions, 399 operand, select.computation(), window_dimensions, window_strides, padding, 481 tensorflow::gtl::ArraySlice<int64> window_dimensions, 485 operand, init_value, local_computation.computation(), window_dimensions,
local_computation_builder.h	188 tensorflow::gtl::ArraySlice<int64> window_dimensions, 245 tensorflow::gtl::ArraySlice<int64> window_dimensions,
xla_client.py	713 def SelectAndScatter(self, operand, select, window_dimensions, window_strides, 722 window_dimensions: sequence of N integers for dimensions of the window. 735 window_dimensions, window_strides) 739 window_dimensions, window_strides, pads, 894 window_dimensions, window_strides, padding): 901 window_dimensions: dimensions of window (sequence of integers). 909 padding, self.GetShape(operand).dimensions(), window_dimensions, 916 window_dimensions, window_strides, pads))     [all...]
xla_client_test.py	[all...]
/external/tensorflow/tensorflow/compiler/xla/tests/
reduce_window_test.cc	71 tensorflow::gtl::ArraySlice<int64> window_dimensions, 78 window_dimensions, window_strides, padding); 82 tensorflow::gtl::ArraySlice<int64> window_dimensions, 86 builder_.ReduceWindow(input, init, CreateScalarMax(), window_dimensions, 91 tensorflow::gtl::ArraySlice<int64> window_dimensions, 97 window_dimensions, window_strides, padding); 111 /*window_dimensions=*/{1, 2}, 127 /*window_dimensions=*/{}, 144 ReduceWindowMin(input, /*window_dimensions=*/{3}, /*window_strides=*/{1}, 277 /*window_dimensions=*/{1, 1, 2, 1}     [all...]
select_and_scatter_test.cc	45 tensorflow::gtl::ArraySlice<int64> window_dimensions; member in struct:xla::__anon39105::SelectAndScatterTestParam 84 operand, ge_f32_, GetParam().window_dimensions, GetParam().window_strides, 203 builder_.SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3}, 215 builder_.SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3}, 226 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3}, 237 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3}, 249 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3}, 263 builder_.SelectAndScatter(reshape, ge_s32_, /*window_dimensions=*/{2, 3}, 275 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3}, 287 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 2}     [all...]

Completed in 156 milliseconds