Home
last modified time | relevance | path

Searched refs:s32_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc39 const Shape s32_ = ShapeUtil::MakeShape(S32, {}); member in xla::__anon8d0540a50111::ShapeInferenceTest
114 Shape tuple = ShapeUtil::MakeTupleShape({s32_, f32_}); in TEST_F()
147 HloOpcode::kSelect, s32_, matrix_64_48_, matrix_64_48_); in TEST_F()
235 ShapeInference::InferTernaryOpShape(HloOpcode::kClamp, s32_, f32_, f32_) in TEST_F()
238 ShapeInference::InferTernaryOpShape(HloOpcode::kClamp, f32_, s32_, f32_) in TEST_F()
241 ShapeInference::InferTernaryOpShape(HloOpcode::kClamp, f32_, f32_, s32_) in TEST_F()
272 ASSERT_FALSE(complex_shape(s32_, s32_, {}).ok()); in TEST_F()
308 ShapeInference::InferVariadicOpShape(HloOpcode::kTuple, {&s32_, &f32_}); in TEST_F()
311 ShapeUtil::MakeTupleShape({s32_, f32_}))); in TEST_F()
944 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F()
[all …]
Dbuffer_assignment_test.cc306 Shape s32_ = ShapeUtil::MakeShape(xla::S32, {}); member in xla::__anon3198ac380111::BufferAssignmentTest
312 Shape t_s32_f32v4_ = ShapeUtil::MakeTupleShape({s32_, f32vec4_});
313 Shape t_s32_f32v10_ = ShapeUtil::MakeTupleShape({s32_, f32vec10_});
780 ShapeUtil::MakeTupleShape({s32_, f32vec10_color1}); in TEST_F()
786 HloInstruction::CreateGetTupleElement(s32_, cond_param, 0)); in TEST_F()
799 HloInstruction::CreateGetTupleElement(s32_, body_param, 0)); in TEST_F()
812 s32_, HloOpcode::kAdd, body_iter, body_iter_increment)); in TEST_F()
820 HloInstruction::CreateParameter(0, s32_, "param_iter")); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc236 HloInstruction::CreateGetTupleElement(s32_, while0, 0)); in CreateTestModule()
244 Shape s32_ = ShapeUtil::MakeShape(xla::S32, {}); member in xla::__anonf885129d0111::BufferDonationTest
247 Shape t_s32_f32v1_ = ShapeUtil::MakeTupleShape({s32_, f32v1_});