Searched refs:r0bool (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conditional_test.cc | 343 Shape r0bool = ShapeUtil::MakeShape(PRED, {}); in XLA_TEST_F() local
345 auto pred_cond = Parameter(&inner_builder, 0, r0bool, "param0"); in XLA_TEST_F()
592 Shape r0bool = ShapeUtil::MakeShape(PRED, {}); in XLA_TEST_F() local
612 Shape r0bool = ShapeUtil::MakeShape(PRED, {}); in XLA_TEST_F() local
634 Shape r0bool = ShapeUtil::MakeShape(PRED, {}); in XLA_TEST_F() local
635 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F()
664 Shape r0bool = ShapeUtil::MakeShape(PRED, {}); in XLA_TEST_F() local
665 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F()
|