| D | select_and_scatter_test.cc | 57 ge_f32_ = CreateScalarGeComputation(F32, &builder_); in SelectAndScatterTest()
66 XlaComputation ge_f32_; member in xla::__anonbb2e76af0111::SelectAndScatterTest
83 SelectAndScatter(operand, ge_f32_, GetParam().window_dimensions, in XLA_TEST_P()
208 SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3}, in XLA_TEST_F()
220 SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3}, in XLA_TEST_F()
269 SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3, 3}, in XLA_TEST_F()
334 SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{2, 2}, in XLA_TEST_F()
359 SelectAndScatter(operand, ge_f32_, {2, 3, 1, 1}, {2, 3, 1, 1}, in TEST_F()
384 SelectAndScatter(operand, ge_f32_, {2, 3, 1, 1}, {2, 2, 1, 1}, in TEST_F()
409 SelectAndScatter(operand, ge_f32_, {2, 3, 1, 1}, {2, 2, 1, 1}, in TEST_F()
[all …]
|