Searched refs:s32_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 37 const Shape s32_ = ShapeUtil::MakeShape(S32, {}); member in xla::__anon8d0540a50111::ShapeInferenceTest
112 Shape tuple = ShapeUtil::MakeTupleShape({s32_, f32_}); in TEST_F()
142 HloOpcode::kSelect, s32_, matrix_64_48_, matrix_64_48_); in TEST_F()
223 ShapeInference::InferTernaryOpShape(HloOpcode::kClamp, s32_, f32_, f32_) in TEST_F()
226 ShapeInference::InferTernaryOpShape(HloOpcode::kClamp, f32_, s32_, f32_) in TEST_F()
229 ShapeInference::InferTernaryOpShape(HloOpcode::kClamp, f32_, f32_, s32_) in TEST_F()
260 ASSERT_FALSE(complex_shape(s32_, s32_, {}).ok()); in TEST_F()
296 ShapeInference::InferVariadicOpShape(HloOpcode::kTuple, {&s32_, &f32_}); in TEST_F()
299 ShapeUtil::MakeTupleShape({s32_, f32_}))); in TEST_F()
567 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F()
[all …]
|
| D | buffer_assignment_test.cc | 299 Shape s32_ = ShapeUtil::MakeShape(xla::S32, {}); member in xla::__anon3198ac380111::BufferAssignmentTest
305 Shape t_s32_f32v4_ = ShapeUtil::MakeTupleShape({s32_, f32vec4_});
306 Shape t_s32_f32v10_ = ShapeUtil::MakeTupleShape({s32_, f32vec10_});
|