Searched refs:MatrixTriangularSolveOp (Results 1 – 3 of 3) sorted by relevance
27 class MatrixTriangularSolveOp : public XlaOpKernel { class29 explicit MatrixTriangularSolveOp(OpKernelConstruction* ctx) in MatrixTriangularSolveOp() function in tensorflow::__anon89dd8abf0111::MatrixTriangularSolveOp82 MatrixTriangularSolveOp::Broadcast(xla::XlaOp lhs, const TensorShape& lhs_shape, in Broadcast()109 REGISTER_XLA_OP(Name("MatrixTriangularSolve"), MatrixTriangularSolveOp);
214 class MatrixTriangularSolveOp217 explicit MatrixTriangularSolveOp(OpKernelConstruction* context)220 ~MatrixTriangularSolveOp() override {}248 MatrixTriangularSolveOp<CPUDevice, TYPE>); \252 MatrixTriangularSolveOp<CPUDevice, TYPE>);427 MatrixTriangularSolveOp<GPUDevice, TYPE>); \431 MatrixTriangularSolveOp<GPUDevice, TYPE>);
183 TypeID::get<TF::MatrixTriangularSolveOp>(), in IsOpAllowedTf2XlaFallback()