Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion_test.cc62 auto scalar_s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
64 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
71 HloInstruction::CreateGetTupleElement(scalar_s32, param, 0)); in TEST_F()
73 HloInstruction::CreateGetTupleElement(scalar_s32, param, 1)); in TEST_F()
76 scalar_s32, HloOpcode::kAdd, gte_0, gte_1)); in TEST_F()
104 auto scalar_s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
106 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32, scalar_s32}); in TEST_F()
113 HloInstruction::CreateGetTupleElement(scalar_s32, param, 0)); in TEST_F()
115 HloInstruction::CreateGetTupleElement(scalar_s32, param, 1)); in TEST_F()
117 HloInstruction::CreateGetTupleElement(scalar_s32, param, 2)); in TEST_F()
[all …]
Dwhile_loop_simplifier_test.cc430 auto scalar_s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
433 ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}))) in TEST_F()
Dpattern_matcher_test.cc775 auto scalar_s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
778 scalar_s32, HloOpcode::kAdd, in TEST_F()
805 scalar_s32, HloOpcode::kAdd, in TEST_F()
806 HloInstruction::CreateParameter(0, scalar_s32, "p").get(), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc28 Shape scalar_s32 = ShapeUtil::MakeShape(S32, {}); in BuildComputation() local
29 XlaOp infeed = InfeedWithToken(CreateToken(&b), scalar_s32); in BuildComputation()
33 GetTupleElement(infeed, 1), scalar_s32, "")); in BuildComputation()
Dwhile_test.cc1191 auto scalar_s32 = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F() local
1193 {scalar_s32, matrix_shape, matrix_shape, matrix_shape}); in XLA_TEST_F()