Searched refs:ROCmSolver (Results 1 – 3 of 3) sorted by relevance
134 ROCmSolver::ROCmSolver(OpKernelContext* context) : context_(context) { in ROCmSolver() function in tensorflow::ROCmSolver159 ROCmSolver::~ROCmSolver() { in ~ROCmSolver()186 Status ROCmSolver::allocate_scoped_tensor(DataType type, in allocate_scoped_tensor()196 Status ROCmSolver::forward_input_or_allocate_scoped_tensor( in forward_input_or_allocate_scoped_tensor()229 Status ROCmSolver::Trsm<Scalar>( \
67 class ROCmSolver {70 explicit ROCmSolver(OpKernelContext* context);71 virtual ~ROCmSolver();94 TF_DISALLOW_COPY_AND_ASSIGN(ROCmSolver);
338 auto solver = absl::make_unique<ROCmSolver>(context);