forward_input_or_allocate_scoped_tensor (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A D	determinant_op.cc	`175 solver->forward_input_or_allocate_scoped_tensor( in ComputeAsync() 320 solver->forward_input_or_allocate_scoped_tensor( in ComputeAsync()`
H A D	self_adjoint_eig_v2_op_gpu.cc	`101 solver->forward_input_or_allocate_scoped_tensor( in ComputeAsync()`
H A D	matrix_solve_op.cc	`197 solver->forward_input_or_allocate_scoped_tensor( in ComputeAsync()`
H A D	svd_op_gpu.cu.cc	`332 solver->forward_input_or_allocate_scoped_tensor( in PerformSVD_MlessN()`
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A D	gpu_solvers.h	`262 Status forward_input_or_allocate_scoped_tensor(`
H A D	rocm_solvers.cc	`720 Status GpuSolver::forward_input_or_allocate_scoped_tensor( in forward_input_or_allocate_scoped_tensor() function in tensorflow::GpuSolver`
H A D	cuda_solvers.cc	`303 Status GpuSolver::forward_input_or_allocate_scoped_tensor( in forward_input_or_allocate_scoped_tensor() function in tensorflow::GpuSolver`