Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
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()
Dsvd_op_gpu.cu.cc117 context, solver->allocate_scoped_tensor(U->dtype(), u_shape, &u_copy), in RunSVD()
143 solver->allocate_scoped_tensor( in RunSVD()
252 solver->allocate_scoped_tensor(M.dtype(), input_shape, &input_copy), in PerformSVD_MgeqN()
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()
Ddeterminant_op.cc189 solver->allocate_scoped_tensor(DataTypeToEnum<int>::value, in ComputeAsync()
334 solver->allocate_scoped_tensor(DataTypeToEnum<int>::value, in ComputeAsync()
Dself_adjoint_eig_v2_op_gpu.cc93 solver->allocate_scoped_tensor(DataTypeToEnum<RealScalar>::value, in ComputeAsync()
Dlu_op_gpu.cu.cc162 solver->allocate_scoped_tensor( in ComputeAsync()
Dcuda_solvers.h178 Status allocate_scoped_tensor(DataType type, const TensorShape& shape,
Dcuda_solvers.cc273 Status CudaSolver::allocate_scoped_tensor(DataType type, in allocate_scoped_tensor() function in tensorflow::CudaSolver