Home
last modified time | relevance | path

Searched refs:pivots (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op_gpu.cu.cc40 int64 num_rows, const int* __restrict__ pivots, in ComputePermutationFromTranspositions() argument
55 permutation_indices[i] = permutation_indices[pivots[i] - 1]; in ComputePermutationFromTranspositions()
56 permutation_indices[pivots[i] - 1] = t; in ComputePermutationFromTranspositions()
162 Tensor pivots; in ComputeAsync() local
166 TensorShape{batch_size, num_rows}, &pivots), in ComputeAsync()
168 auto pivots_mat = pivots.template matrix<int32>(); in ComputeAsync()
224 int* pivots_ptr = pivots.flat<int>().data(); in ComputeAsync()
Ddeterminant_op_gpu.cu.cc33 __device__ int PermutationOrder(int n, const int* __restrict__ pivots) { in PermutationOrder() argument
41 order += pivots[i] != (i + 1); in PermutationOrder()
126 const int* pivots, typename TTypes<Scalar, 1>::Tensor output, in operator ()()
135 config.virtual_thread_count, n, lu_factor.data(), pivots, nullptr, in operator ()()
149 const int* pivots, typename TTypes<Scalar, 1>::Tensor sign, in operator ()()
157 config.virtual_thread_count, n, lu_factor.data(), pivots, sign.data(), in operator ()()
Ddeterminant_op.h30 const int* pivots, typename TTypes<Scalar, 1>::Tensor output,
40 const int* pivots, typename TTypes<Scalar, 1>::Tensor sign,
Ddeterminant_op.cc186 Tensor pivots; in ComputeAsync() local
190 TensorShape{batch_size, n}, &pivots), in ComputeAsync()
192 auto pivots_mat = pivots.template matrix<int>(); in ComputeAsync()
331 Tensor pivots; in ComputeAsync() local
335 TensorShape{batch_size, n}, &pivots), in ComputeAsync()
337 auto pivots_mat = pivots.template matrix<int>(); in ComputeAsync()
Dmatrix_inverse_op.cc151 Tensor pivots; in ComputeAsync() local
155 TensorShape{batch_size, n}, &pivots), in ComputeAsync()
157 auto pivots_mat = pivots.template matrix<int>(); in ComputeAsync()
Dmatrix_solve_op.cc203 Tensor pivots; in ComputeAsync() local
207 TensorShape{batch_size, n}, &pivots), in ComputeAsync()
209 auto pivots_mat = pivots.template matrix<int>(); in ComputeAsync()
/external/ImageMagick/MagickCore/
Dmatrix.c506 *pivots, in GaussJordanElimination() local
512 pivots=(ssize_t *) AcquireQuantumMemory(rank,sizeof(*pivots)); in GaussJordanElimination()
514 (pivots == (ssize_t *) NULL)) in GaussJordanElimination()
516 if (pivots != (ssize_t *) NULL) in GaussJordanElimination()
517 pivots=(ssize_t *) RelinquishMagickMemory(pivots); in GaussJordanElimination()
526 (void) memset(pivots,0,rank*sizeof(*pivots)); in GaussJordanElimination()
533 if (pivots[j] != 1) in GaussJordanElimination()
536 if (pivots[k] != 0) in GaussJordanElimination()
538 if (pivots[k] > 1) in GaussJordanElimination()
549 pivots[column]++; in GaussJordanElimination()
[all …]
/external/eigen/Eigen/src/LU/
DFullPivLU.h658 Matrix<Index, Dynamic, 1, 0, MaxSmallDimAtCompileTime, 1> pivots(rank());
663 pivots.coeffRef(p++) = i;
676 m.row(i).tail(cols-i) = dec().matrixLU().row(pivots.coeff(i)).tail(cols-i);
681 m.col(i).swap(m.col(pivots.coeff(i)));
693 m.col(i).swap(m.col(pivots.coeff(i)));
727 Matrix<Index, Dynamic, 1, 0, MaxSmallDimAtCompileTime, 1> pivots(rank());
732 pivots.coeffRef(p++) = i;
736 dst.col(i) = originalMatrix().col(dec().permutationQ().indices().coeff(pivots.coeff(i)));
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_case_op.cc227 std::vector<NodeOut> pivots(num_branches_); in AddOutputs() local
229 pivots[j] = NodeOut(pivots_[j]); in AddOutputs()
233 .Input(pivots) in AddOutputs()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dlu_decomposition.h51 XlaOp pivots; member
/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc486 const Scalar* A, int lda, const int* pivots, in GetrsImpl() argument
491 CUDAComplex(A), lda, pivots, in GetrsImpl()
500 const int* pivots, Scalar* B, int ldb, int* dev_lapack_info) const { \
502 cusolver_dn_handle_, trans, n, nrhs, A, lda, pivots, B, \
Dcuda_solvers.h245 int lda, const int* pivots, Scalar* B, int ldb,
/external/eigen/doc/
DCustomizingEigen_CustomScalar.dox62 …of a number, but we can define a different score, for instance to prefer pivots with a more compac…
/external/tensorflow/tensorflow/compiler/xla/python/
Dops.cc203 return std::make_tuple(lu.lu, lu.pivots, lu.permutation); in BuildOpsSubmodule()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc2887 absl::node_hash_map<string, Node*> pivots; in ProcessHeadTailOutsideCompilation() local
2891 pivots[cluster_name] = node; in ProcessHeadTailOutsideCompilation()
2896 Node* pivot_node = pivots[iter.first]; in ProcessHeadTailOutsideCompilation()
/external/jline/src/src/test/resources/jline/example/
Denglish.gz
/external/icu/icu4j/main/shared/data/
DTransliterator_Han_Latin_Definition.txt12161 椳 < the\-pivots;
44195 椳 > the\-pivots;
/external/kotlinx.coroutines/benchmarks/src/jmh/resources/
Dospd.txt.gz
/external/cldr/tools/java/org/unicode/cldr/util/data/transforms/
Dinternal_raw_IPA-old.txt151248 pivots %15379