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

Completed in 1086 milliseconds