Home
last modified time | relevance | path

Searched refs:forward_input_or_allocate_scoped_tensor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Drocm_solvers.h77 Status forward_input_or_allocate_scoped_tensor(
Drocm_solvers.cc196 Status ROCmSolver::forward_input_or_allocate_scoped_tensor( in forward_input_or_allocate_scoped_tensor() function in tensorflow::ROCmSolver
Dcuda_solvers.h187 Status forward_input_or_allocate_scoped_tensor(
Dcuda_solvers.cc304 Status CudaSolver::forward_input_or_allocate_scoped_tensor( in forward_input_or_allocate_scoped_tensor() function in tensorflow::CudaSolver
/external/tensorflow/tensorflow/core/kernels/linalg/
Dself_adjoint_eig_v2_op_gpu.cc101 solver->forward_input_or_allocate_scoped_tensor( in ComputeAsync()
Ddeterminant_op.cc175 solver->forward_input_or_allocate_scoped_tensor( in ComputeAsync()
320 solver->forward_input_or_allocate_scoped_tensor( in ComputeAsync()
Dmatrix_solve_op.cc190 solver->forward_input_or_allocate_scoped_tensor( in ComputeAsync()
Dsvd_op_gpu.cu.cc328 solver->forward_input_or_allocate_scoped_tensor( in PerformSVD_MlessN()