Home
last modified time | relevance | path

Searched refs:allocate_scoped_tensor (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dsvd_op_gpu.cu.cc152 context, solver->allocate_scoped_tensor(U->dtype(), u_shape, &u_copy), in RunSVD()
157 solver->allocate_scoped_tensor(V->dtype(), v_shape, &v_copy), done); in RunSVD()
177 solver->allocate_scoped_tensor( in RunSVD()
303 solver->allocate_scoped_tensor(M.dtype(), input_shape, &input_copy), in PerformSVD_MgeqN()
Dmatrix_solve_op.cc182 solver->allocate_scoped_tensor(DataTypeToEnum<Scalar>::value, in ComputeAsync()
206 solver->allocate_scoped_tensor(DataTypeToEnum<int>::value, in ComputeAsync()
258 solver->allocate_scoped_tensor(DataTypeToEnum<Scalar>::value, in ComputeAsync()
Dmatrix_inverse_op.cc135 solver->allocate_scoped_tensor(DataTypeToEnum<Scalar>::value, in ComputeAsync()
154 solver->allocate_scoped_tensor(DataTypeToEnum<int>::value, in ComputeAsync()
Dqr_op_impl.h184 solver->allocate_scoped_tensor(DataTypeToEnum<Scalar>::value, in ComputeAsync()
190 solver->allocate_scoped_tensor( in ComputeAsync()
Dtridiagonal_solve_op_gpu.cu.cc174 cublas_solver->allocate_scoped_tensor( in ComputeMatrix()
349 OP_REQUIRES_OK(context, cublas_solver->allocate_scoped_tensor( in TransposeLhsForGtsvBatched()
Dself_adjoint_eig_v2_op_gpu.cc93 solver->allocate_scoped_tensor(DataTypeToEnum<RealScalar>::value, in ComputeAsync()
Ddeterminant_op.cc189 solver->allocate_scoped_tensor(DataTypeToEnum<int>::value, in ComputeAsync()
334 solver->allocate_scoped_tensor(DataTypeToEnum<int>::value, in ComputeAsync()
Dlu_op_gpu.cu.cc164 solver->allocate_scoped_tensor( in ComputeAsync()
/external/tensorflow/tensorflow/core/util/
Drocm_solvers.h75 Status allocate_scoped_tensor(DataType type, const TensorShape& shape,
Drocm_solvers.cc186 Status ROCmSolver::allocate_scoped_tensor(DataType type, in allocate_scoped_tensor() function in tensorflow::ROCmSolver
Dcuda_solvers.h185 Status allocate_scoped_tensor(DataType type, const TensorShape& shape,
Dcuda_solvers.cc294 Status CudaSolver::allocate_scoped_tensor(DataType type, in allocate_scoped_tensor() function in tensorflow::CudaSolver