Home
last modified time | relevance | path

Searched refs:InferCholeskyShape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h126 static StatusOr<Shape> InferCholeskyShape(const Shape& a);
Dhlo_verifier.cc179 TF_ASSIGN_OR_RETURN(const Shape expected, ShapeInference::InferCholeskyShape( in HandleCholesky()
Dshape_inference.cc2013 /* static */ StatusOr<Shape> ShapeInference::InferCholeskyShape( in InferCholeskyShape() function in xla::ShapeInference
Dhlo_parser.cc1731 return ShapeInference::InferCholeskyShape(operands[0]->shape()); in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc4444 ShapeInference::InferCholeskyShape(*a_shape)); in Cholesky()