Searched refs:ge_s32_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_and_scatter_test.cc | 55 ge_s32_ = CreateScalarGeComputation(S32, &builder_); in SelectAndScatterTest()
64 XlaComputation ge_s32_; member in xla::__anonbb2e76af0111::SelectAndScatterTest
231 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3}, in XLA_TEST_F()
242 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3}, in XLA_TEST_F()
254 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3}, in XLA_TEST_F()
283 SelectAndScatter(reshape, ge_s32_, /*window_dimensions=*/{2, 3}, in XLA_TEST_F()
295 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3}, in XLA_TEST_F()
307 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 2}, in XLA_TEST_F()
321 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 2}, in XLA_TEST_F()
|