Searched refs:input_global_data (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_test.cc | 108 std::unique_ptr<GlobalData> input_global_data = in RunR1ToR0Test() local
115 ComputeAndCompareR0<float>(&builder, expected, {input_global_data.get()}, in RunR1ToR0Test()
139 std::unique_ptr<GlobalData> input_global_data = in RunR1ToR0PredTest() local
150 ComputeAndCompareR0<bool>(&builder, expected, {input_global_data.get()}); in RunR1ToR0PredTest()
182 std::unique_ptr<GlobalData> input_global_data = in RunR2ToR1PredTest() local
198 ComputeAndCompareR1<bool>(&builder, expected, {input_global_data.get()}); in RunR2ToR1PredTest()
215 std::unique_ptr<GlobalData> input_global_data = in RunR2ToR0Test() local
224 ComputeAndCompareR0<float>(&builder, expected, {input_global_data.get()}, in RunR2ToR0Test()
242 std::unique_ptr<GlobalData> input_global_data = in RunR2ToR1Test() local
253 ComputeAndCompareR1<float>(&builder, expected, {input_global_data.get()}, in RunR2ToR1Test()
[all …]
|