Searched refs:unpack_data (Results 1 – 8 of 8) sorted by relevance
| /external/mesa3d/src/compiler/nir/ |
| D | nir_range_analysis.c | 137 unpack_data(uint32_t v) in unpack_data() function
748 const struct ssa_result_range left = unpack_data(src_res[0]); in process_fp_query()
749 const struct ssa_result_range right = unpack_data(src_res[1]); in process_fp_query()
774 r = unpack_data(src_res[0]); in process_fp_query()
786 r = unpack_data(src_res[0]); in process_fp_query()
808 const struct ssa_result_range left = unpack_data(src_res[0]); in process_fp_query()
809 const struct ssa_result_range right = unpack_data(src_res[1]); in process_fp_query()
833 r = unpack_data(src_res[0]); in process_fp_query()
848 const struct ssa_result_range left = unpack_data(src_res[0]); in process_fp_query()
849 const struct ssa_result_range right = unpack_data(src_res[1]); in process_fp_query()
[all …]
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesLinearAlgebra.cpp | 271 bool unpack_data, bool unpack_pivots) { in linalg_lu_unpack_batch_rule() argument
286 const auto res = at::lu_unpack(LU_, pivots_, unpack_data, unpack_pivots); in linalg_lu_unpack_batch_rule()
|
| /external/pytorch/torch/csrc/inductor/aoti_torch/generated/ |
| D | c_shim_cpu.h | 78 …u_unpack(AtenTensorHandle LU_data, AtenTensorHandle LU_pivots, int32_t unpack_data, int32_t unpack…
|
| D | c_shim_cuda.h | 87 …u_unpack(AtenTensorHandle LU_data, AtenTensorHandle LU_pivots, int32_t unpack_data, int32_t unpack…
|
| /external/pytorch/aten/src/ATen/native/ |
| D | BatchLinearAlgebra.cpp | 695 TORCH_META_FUNC(lu_unpack)(const Tensor& LU, const Tensor& pivots, bool unpack_data, bool unpack_pi… in TORCH_META_FUNC()
716 if (unpack_data) { in TORCH_META_FUNC()
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 1209 unpack_data: bool = True,
1233 if unpack_data:
|
| D | overrides.py | 652 torch.lu_unpack: lambda LU_data, LU_pivots, unpack_data=True, unpack_pivots=True: -1,
|
| /external/pytorch/test/ |
| D | test_linalg.py | 5634 p, l, u = torch.lu_unpack(lu_data, lu_pivots, unpack_data=False)
5638 p, l, u = torch.lu_unpack(lu_data, lu_pivots, unpack_data=False, unpack_pivots=False)
|