| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebraKernel.cpp | 756 const Tensor& pivots, in apply_ldl_factor()
808 const Tensor& pivots, in ldl_factor_kernel()
821 const Tensor& pivots, in apply_ldl_solve()
874 const Tensor& pivots, in ldl_solve_kernel()
898 void apply_lu_factor(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_p… in apply_lu_factor()
943 void lu_factor_kernel(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_… in lu_factor_kernel()
963 void apply_lu_solve(const Tensor& LU, const Tensor& pivots, const Tensor& B, TransposeType transpos… in apply_lu_solve()
1009 void lu_solve_kernel(const Tensor& LU, const Tensor& pivots, const Tensor& B, TransposeType trans) { in lu_solve_kernel()
|
| H A D | BatchLinearAlgebra.cpp | 1913 auto pivots = B.new_empty({0}, kInt); in linalg_solve_ex_out() local
1974 …e<Tensor&, Tensor&> linalg_lu_factor_out(const Tensor& A, bool pivot, Tensor& LU, Tensor& pivots) { in linalg_lu_factor_out()
2024 auto pivots = at::empty({0}, A.options().dtype(kInt)); in TORCH_IMPL_FUNC() local
3729 Tensor& pivots) { in linalg_ldl_factor_out()
|
| H A D | LinearAlgebra.cpp | 350 static Tensor lu_det_P(const Tensor& pivots) { in lu_det_P()
378 auto pivots = at::empty({0}, A.options().dtype(kInt)); in linalg_det_out() local
414 auto pivots = at::empty({0}, A.options().dtype(kInt)); in linalg_slogdet_out() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebra.cpp | 1040 const Tensor& pivots, in apply_ldl_factor_magma()
1084 const Tensor& pivots, in ldl_factor_magma()
1104 const Tensor& pivots, in ldl_factor_kernel()
1136 const Tensor& pivots, in ldl_solve_kernel()
1476 static void apply_lu_factor_looped_magma(const Tensor& input, const Tensor& pivots, const Tensor& i… in apply_lu_factor_looped_magma()
1530 static void apply_lu_factor_batched_magma(const Tensor& input, const Tensor& pivots, const Tensor& … in apply_lu_factor_batched_magma()
1589 static void lu_factor_looped_magma(const Tensor& input, const Tensor& pivots, const Tensor& infos, … in lu_factor_looped_magma()
1595 static void lu_factor_batched_magma(const Tensor& input, const Tensor& pivots, const Tensor& infos,… in lu_factor_batched_magma()
1601 static void lu_factor(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_… in lu_factor()
1607 …const auto lu_factor_magma = [batch_size](const Tensor& input, const Tensor& pivots, const Tensor&… in lu_factor()
[all …]
|
| H A D | BatchLinearAlgebraLibBlas.cpp | 108 static void apply_lu_factor_batched_cublas(const Tensor& A, const Tensor& pivots, const Tensor& inf… in apply_lu_factor_batched_cublas()
124 void lu_factor_batched_cublas(const Tensor& A, const Tensor& pivots, const Tensor& infos, bool get_… in lu_factor_batched_cublas()
131 static void apply_lu_solve_batched_cublas(const Tensor& LU, const Tensor& pivots, const Tensor& B, … in apply_lu_solve_batched_cublas()
154 void lu_solve_batched_cublas(const Tensor& LU, const Tensor& pivots, const Tensor& B, TransposeType… in lu_solve_batched_cublas()
|
| H A D | BatchLinearAlgebraLib.cpp | 66 const Tensor& pivots, in apply_ldl_factor_cusolver()
115 const Tensor& pivots, in apply_ldl_solve_cusolver()
210 const Tensor& pivots, in ldl_factor_cusolver()
230 const Tensor& pivots, in ldl_solve_cusolver()
1458 void lu_factor_looped_cusolver(const Tensor& self, const Tensor& pivots, const Tensor& infos, bool … in lu_factor_looped_cusolver()
1503 void lu_solve_looped_cusolver(const Tensor& LU, const Tensor& pivots, const Tensor& B, TransposeTyp… in lu_solve_looped_cusolver()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LinearAlgebraStubs.cpp | 63 void lazy_lu_factor(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pi… in lazy_lu_factor()
110 void lazy_lu_solve(const Tensor& LU, const Tensor& pivots, const Tensor& B, TransposeType trans) { in lazy_lu_solve()
122 const Tensor& pivots, in lazy_ldl_factor()
132 const Tensor& pivots, in lazy_ldl_solve()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | lu_op_gpu.cu.cc | 40 int64 num_rows, const int* __restrict__ pivots, in ComputePermutationFromTranspositions()
162 Tensor pivots; in ComputeAsync() local
|
| H A D | determinant_op.cc | 186 Tensor pivots; in ComputeAsync() local
331 Tensor pivots; in ComputeAsync() local
|
| H A D | determinant_op_gpu.cu.cc | 34 __device__ int PermutationOrder(int n, const int* __restrict__ pivots) { in PermutationOrder()
|
| H A D | matrix_inverse_op.cc | 151 Tensor pivots; in ComputeAsync() local
|
| H A D | matrix_solve_op.cc | 210 Tensor pivots; in ComputeAsync() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | linalg.h | 97 Tensor& pivots, in lu_factor_out()
882 Tensor& pivots, in ldl_factor_ex_out()
896 const Tensor& pivots, in ldl_solve()
905 const Tensor& pivots, in ldl_solve_out()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | lu_decomposition.h | 51 XlaOp pivots; member
|
| /aosp_15_r20/external/perfetto/ui/src/core/ |
| H A D | pivot_table_query_generator.ts | 129 const pivots = state.selectedPivots; constant
|
| H A D | pivot_table_manager.ts | 237 const pivots = this.state.selectedPivots; constant
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | lower_case_op.cc | 227 std::vector<NodeOut> pivots(num_branches_); in AddOutputs() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesLinearAlgebra.cpp | 270 const Tensor& pivots, std::optional<int64_t> pivots_bdim, in linalg_lu_unpack_batch_rule()
292 const Tensor& pivots, std::optional<int64_t> pivots_bdim, in linalg_lu_solve_batch_rule()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 4188 const Tensor& pivots, in linalg_det_jvp()
4207 const Tensor& pivots) { in linalg_det_backward()
4272 const Tensor& pivots, in slogdet_jvp()
4296 const Tensor& pivots) { in slogdet_backward()
5700 const Tensor& pivots, in linalg_lu_solve_LU()
5769 const Tensor& pivots, in linalg_lu_solve_jvp()
5839 const Tensor& pivots, in linalg_solve_jvp()
5873 const Tensor& pivots, in linalg_solve_backward()
|
| /aosp_15_r20/external/google-cloud-java/java-analytics-data/proto-google-analytics-data-v1beta/src/main/proto/google/analytics/data/v1beta/ |
| H A D | analytics_data_api.proto | 362 repeated Pivot pivots = 5; field
|
| /aosp_15_r20/external/googleapis/google/analytics/data/v1beta/ |
| H A D | analytics_data_api.proto | 484 repeated Pivot pivots = 5; field
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | tensorexpr_init.cpp | 585 [](const ForPtr& f, const std::unordered_set<StmtPtr>& pivots) { in initTensorExprBindings()
|
| H A D | loopnest.cpp | 1681 const std::unordered_set<StmtPtr>& pivots) { in distributeLoop()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | cuda_solvers.cc | 483 const Scalar* A, int lda, const int* pivots, in GetrsImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | encapsulate_tpu_computations_pass.cc | 2835 absl::node_hash_map<string, Node*> pivots; in ProcessHeadTailOutsideCompilation() local
|