Searched defs:pivots (Results 1 – 10 of 10) sorted by relevance
51 XlaOp pivots; member
40 int64 num_rows, const int* __restrict__ pivots, in ComputePermutationFromTranspositions()162 Tensor pivots; in ComputeAsync() local
186 Tensor pivots; in ComputeAsync() local331 Tensor pivots; in ComputeAsync() local
151 Tensor pivots; in ComputeAsync() local
203 Tensor pivots; in ComputeAsync() local
33 __device__ int PermutationOrder(int n, const int* __restrict__ pivots) { in PermutationOrder()
227 std::vector<NodeOut> pivots(num_branches_); in AddOutputs() local
506 *pivots, in GaussJordanElimination() local
486 const Scalar* A, int lda, const int* pivots, in GetrsImpl()
2887 absl::node_hash_map<string, Node*> pivots; in ProcessHeadTailOutsideCompilation() local