Searched defs:window_dimensions (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
H A Dpadding.cc28 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];
26 ValidatePaddingValues( tensorflow::gtl::ArraySlice<int64> input_dimensions, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides) argument
42 MakePadding( tensorflow::gtl::ArraySlice<int64> input_dimensions, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, Padding padding) argument
H A Dcomputation_builder.cc130 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(
129 MakeWindow( tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, tensorflow::gtl::ArraySlice<std::pair<int64, int64>> padding, tensorflow::gtl::ArraySlice<int64> lhs_dilation, tensorflow::gtl::ArraySlice<int64> rhs_dilation, Window* window) argument
1301 ReduceWindow( const ComputationDataHandle& operand, const ComputationDataHandle& init_value, const Computation& computation, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, Padding padding) argument
1331 ReduceWindowWithGeneralPadding( const ComputationDataHandle& operand, const ComputationDataHandle& init_value, const Computation& computation, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, tensorflow::gtl::ArraySlice<std::pair<int64, int64>> padding) argument
1409 SelectAndScatter( const ComputationDataHandle& operand, const Computation& select, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, Padding padding, const ComputationDataHandle& source, const ComputationDataHandle& init_value, const Computation& scatter) argument
1430 SelectAndScatterWithGeneralPadding( const ComputationDataHandle& operand, const Computation& select, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, tensorflow::gtl::ArraySlice<std::pair<int64, int64>> padding, const ComputationDataHandle& source, const ComputationDataHandle& init_value, const Computation& scatter) argument
[all...]
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dselect_and_scatter_test.cc45 tensorflow::gtl::ArraySlice<int64> window_dimensions; member in struct:xla::__anon25784::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
[all...]
H A Dreduce_window_test.cc71 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
70 ReduceWindowAdd(const ComputationDataHandle& input, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, Padding padding) argument
81 ReduceWindowMax(const ComputationDataHandle& input, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, Padding padding) argument
90 ReduceWindowMin(const ComputationDataHandle& input, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, Padding padding) argument
[all...]
/external/tensorflow/tensorflow/compiler/xla/python/
H A Dlocal_computation_builder.cc393 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,
391 SelectAndScatterWithGeneralPadding( const ComputationDataHandle& operand, const LocalComputation& select, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, tensorflow::gtl::ArraySlice<std::pair<int64, int64>> padding, const ComputationDataHandle& source, const ComputationDataHandle& init_value, const LocalComputation& scatter) argument
477 ReduceWindowWithGeneralPadding( const ComputationDataHandle& operand, const ComputationDataHandle& init_value, const LocalComputation& local_computation, tensorflow::gtl::ArraySlice<int64> window_dimensions, tensorflow::gtl::ArraySlice<int64> window_strides, tensorflow::gtl::ArraySlice<std::pair<int64, int64>> padding) argument

Completed in 101 milliseconds