Home
last modified time | relevance | path

Searched refs:and_reduce (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_test.cc119 void RunR1ToR0PredTest(bool and_reduce, absl::Span<const int> input_data) { in RunR1ToR0PredTest() argument
128 if (and_reduce) { in RunR1ToR0PredTest()
142 bool expected = and_reduce; in RunR1ToR0PredTest()
144 if (and_reduce) { in RunR1ToR0PredTest()
157 void RunR2ToR1PredTest(bool and_reduce, int64 rows, int64 minor = 1, in RunR2ToR1PredTest() argument
166 if (and_reduce) { in RunR2ToR1PredTest()
187 bool column_sum = and_reduce ? true : false; in RunR2ToR1PredTest()
189 if (and_reduce) { in RunR2ToR1PredTest()