Searched refs:complex_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 257 auto complex_shape = [&](const Shape& lhs, const Shape& rhs, in TEST_F() local
263 ASSERT_FALSE(complex_shape(s32_, s32_, {}).ok()); in TEST_F()
264 ASSERT_FALSE(complex_shape(pred_, pred_, {}).ok()); in TEST_F()
266 ASSERT_FALSE(complex_shape(f32_, f64_, {}).ok()); in TEST_F()
268 ASSERT_FALSE(complex_shape(f64_, f64_, {}).ok()); in TEST_F()
271 TF_ASSERT_OK_AND_ASSIGN(Shape result, complex_shape(f32_, f32_, {})); in TEST_F()
273 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {})); in TEST_F()
275 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(f32_, vector_32_, {})); in TEST_F()
277 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {})); in TEST_F()
282 complex_shape(vector_64_, matrix_32_64_, {1})); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 360 const Shape& complex_shape) { in ComplexComponentShape() argument
361 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
362 return ChangeElementType(complex_shape, primitive_util::ComplexComponentType( in ComplexComponentShape()
363 complex_shape.element_type())); in ComplexComponentShape()
|
| D | shape_util.h | 424 static Shape ComplexComponentShape(const Shape& complex_shape);
|