Searched refs:cholesky_op (Results 1 – 5 of 5) sorted by relevance
37 "cholesky_op.cc",68 ":cholesky_op",126 name = "cholesky_op",127 prefix = "cholesky_op",
25 "cholesky_op.cc",
652 TF_ASSIGN_OR_RETURN(auto cholesky_op, in EmitCholesky()656 cholesky_op.is_lowerAttr(builder_.getBoolAttr(options.lower())); in EmitCholesky()657 return cholesky_op; in EmitCholesky()
1486 auto cholesky_op = mlir::cast<mlir::lmhlo_gpu::CholeskyOp>(input.op); in EmitCholeskyThunkFromMlir() local1488 const Shape shape = TypeToShape(cholesky_op.input().getType()); in EmitCholeskyThunkFromMlir()1498 GetAllocationSliceForMlir(cholesky_op.input())); in EmitCholeskyThunkFromMlir()1500 GetAllocationSliceForMlir(cholesky_op.output())); in EmitCholeskyThunkFromMlir()1502 GetAllocationSliceForMlir(cholesky_op.scratch())); in EmitCholeskyThunkFromMlir()1504 GetAllocationSliceForMlir(cholesky_op.info())); in EmitCholeskyThunkFromMlir()1517 options.set_lower(cholesky_op.is_lower()); in EmitCholeskyThunkFromMlir()
6297 "//tensorflow/core/kernels/linalg:cholesky_op.cc",