Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h124 static StatusOr<Shape> InferCholeskyShape(const Shape& a);
Dhlo_verifier.cc182 TF_ASSIGN_OR_RETURN(const Shape expected, ShapeInference::InferCholeskyShape( in HandleCholesky()
Dshape_inference.cc1954 /* static */ StatusOr<Shape> ShapeInference::InferCholeskyShape( in InferCholeskyShape() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3060 ShapeInference::InferCholeskyShape(a_shape)); in Cholesky()