Searched refs:vector_32_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 45 const Shape vector_32_ = ShapeUtil::MakeShape(F32, {32}); member in xla::__anon8d0540a50111::ShapeInferenceTest
233 HloOpcode::kClamp, vector_64_, vector_32_, vector_32_) in TEST_F()
236 HloOpcode::kClamp, vector_32_, vector_64_, vector_32_) in TEST_F()
239 HloOpcode::kClamp, vector_32_, vector_32_, vector_64_) in TEST_F()
243 vector_64_, vector_32_, f32_) in TEST_F()
246 vector_64_, f32_, vector_32_) in TEST_F()
249 vector_64_, vector_32_) in TEST_F()
270 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {})); in TEST_F()
272 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(f32_, vector_32_, {})); in TEST_F()
274 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {})); in TEST_F()
[all …]
|