Searched refs:complex_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 266 auto complex_shape = [&](const Shape& lhs, const Shape& rhs, in TEST_F() local
272 ASSERT_FALSE(complex_shape(s32_, s32_, {}).ok()); in TEST_F()
273 ASSERT_FALSE(complex_shape(pred_, pred_, {}).ok()); in TEST_F()
275 ASSERT_FALSE(complex_shape(f32_, f64_, {}).ok()); in TEST_F()
277 ASSERT_FALSE(complex_shape(f16_, f16_, {}).ok()); in TEST_F()
280 TF_ASSERT_OK_AND_ASSIGN(Shape result, complex_shape(f32_, f32_, {})); in TEST_F()
282 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {})); in TEST_F()
284 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(f32_, vector_32_, {})); in TEST_F()
286 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {})); in TEST_F()
291 complex_shape(vector_64_, matrix_32_64_, {1})); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 529 const Shape& complex_shape) { in ComplexComponentShape() argument
530 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
531 return ChangeElementType(complex_shape, primitive_util::ComplexComponentType( in ComplexComponentShape()
532 complex_shape.element_type())); in ComplexComponentShape()
|
| D | shape_util.h | 546 static Shape ComplexComponentShape(const Shape& complex_shape);
|