HomeSort by relevance Sort by last modified time
    Searched refs:complex_shape (Results 1 - 3 of 3) sorted by null

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

Completed in 109 milliseconds