Searched defs:data_ptrs (Results 1 – 12 of 12) sorted by relevance
369 void* data_ptrs[3]; in run_conv_plan() local416 void* data_ptrs[] = { in run_conv_plan_fused() local710 void* data_ptrs[] = {x.data_ptr(), y.data_ptr(), w.data_ptr()}; in get_plans_from_find() local759 void* data_ptrs[] = { in get_plans_from_find_fused() local
894 std::vector<void*> data_ptrs; in get_parameters() local
38 const T** data_ptrs = GetGpuDeviceArrayOnDevice(&input_ptrs); in DynamicStitchKernel() local
117 std::vector<void*> data_ptrs = {}; in Run() local
107 std::vector<void*> data_ptrs = {}; in Run() local
1269 const uint8_t *const data_ptrs[3] = { in TYPED_TEST_P() local1294 const uint8_t *const data_ptrs[3] = { in TYPED_TEST_P() local1322 const uint8_t *const data_ptrs[3] = { in TYPED_TEST_P() local
140 std::vector<void *> data_ptrs; in add() local
120 at::SmallVector<void *, 7> data_ptrs; in apply_impl_helper() local
148 at::SmallVector<void *, 8> data_ptrs; in apply_impl_helper() local
1254 [](py::list& tensors, py::list& data_ptrs, py::list& indices) { in registerCudaPluggableAllocator()
4464 std::array<void*, sizeof...(Args)> data_ptrs = {inputs.opaque()...}; in operator ()() local
71 std::vector<void*> data_ptrs; in qcat_nhwc_kernel() local