Searched refs:complex_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 254 auto complex_shape = [&](const Shape& lhs, const Shape& rhs, in TEST_F() local
260 ASSERT_FALSE(complex_shape(s32_, s32_, {}).ok()); in TEST_F()
261 ASSERT_FALSE(complex_shape(pred_, pred_, {}).ok()); in TEST_F()
263 ASSERT_FALSE(complex_shape(f32_, f64_, {}).ok()); in TEST_F()
265 ASSERT_FALSE(complex_shape(f16_, f16_, {}).ok()); in TEST_F()
268 TF_ASSERT_OK_AND_ASSIGN(Shape result, complex_shape(f32_, f32_, {})); 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()
279 complex_shape(vector_64_, matrix_32_64_, {1})); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 372 const Shape& complex_shape) { in ComplexComponentShape() argument
373 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
374 return ChangeElementType(complex_shape, primitive_util::ComplexComponentType( in ComplexComponentShape()
375 complex_shape.element_type())); in ComplexComponentShape()
|
| D | shape_util.h | 484 static Shape ComplexComponentShape(const Shape& complex_shape);
|