Searched refs:complex_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 265 auto complex_shape = [&](const Shape& lhs, const Shape& rhs, in TEST_F() local
271 ASSERT_FALSE(complex_shape(s32_, s32_, {}).ok()); in TEST_F()
272 ASSERT_FALSE(complex_shape(pred_, pred_, {}).ok()); in TEST_F()
274 ASSERT_FALSE(complex_shape(f32_, f64_, {}).ok()); in TEST_F()
276 ASSERT_FALSE(complex_shape(f16_, f16_, {}).ok()); in TEST_F()
279 TF_ASSERT_OK_AND_ASSIGN(Shape result, complex_shape(f32_, f32_, {})); in TEST_F()
281 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {})); in TEST_F()
283 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(f32_, vector_32_, {})); in TEST_F()
285 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {})); in TEST_F()
290 complex_shape(vector_64_, matrix_32_64_, {1})); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 419 const Shape& complex_shape) { in ComplexComponentShape() argument
420 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
421 return ChangeElementType(complex_shape, primitive_util::ComplexComponentType( in ComplexComponentShape()
422 complex_shape.element_type())); in ComplexComponentShape()
|
| D | shape_util.h | 497 static Shape ComplexComponentShape(const Shape& complex_shape);
|