Home
last modified time | relevance | path

Searched refs:input_copy_reshaped (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddeterminant_op.cc182 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
183 const int64 batch_size = input_copy_reshaped.dimension(0); in ComputeAsync()
210 input_copy_ptrs_base[batch] = &input_copy_reshaped(batch, 0, 0); in ComputeAsync()
226 solver->Getrf(n, n, &input_copy_reshaped(batch, 0, 0), n, in ComputeAsync()
327 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
328 const int64 batch_size = input_copy_reshaped.dimension(0); in ComputeAsync()
354 input_copy_ptrs_base[batch] = &input_copy_reshaped(batch, 0, 0); in ComputeAsync()
370 solver->Getrf(n, n, &input_copy_reshaped(batch, 0, 0), n, in ComputeAsync()
Dmatrix_inverse_op.cc138 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
149 const int64 batch_size = input_copy_reshaped.dimension(0); in ComputeAsync()
177 input_copy_ptr_array_base[batch] = &input_copy_reshaped(batch, 0, 0); in ComputeAsync()
221 solver->Getrf(n, n, &input_copy_reshaped(batch, 0, 0), n, in ComputeAsync()
235 solver->Getrs(CUBLAS_OP_N, n, n, &input_copy_reshaped(batch, 0, 0), in ComputeAsync()
Dmatrix_solve_op.cc199 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
200 const int64 batch_size = input_copy_reshaped.dimension(0); in ComputeAsync()
226 input_copy_ptrs_base[batch] = &input_copy_reshaped(batch, 0, 0); in ComputeAsync()
242 solver->Getrf(n, n, &input_copy_reshaped(batch, 0, 0), n, in ComputeAsync()
292 input_copy_ptrs_base[batch] = &input_copy_reshaped(batch, 0, 0); in ComputeAsync()
315 &input_copy_reshaped(batch, 0, 0), n, in ComputeAsync()
Dself_adjoint_eig_v2_op_gpu.cc127 auto input_copy_reshaped = input_copy.flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
136 &input_copy_reshaped(batch, 0, 0), n, in ComputeAsync()