Searched defs:perm_data (Results 1 – 8 of 8) sorted by relevance
51 const int* perm_data = GetTensorData<int32_t>(op_context->perm); in ResizeOutputTensor() local99 const int* perm_data = GetTensorData<int32_t>(op_context.perm); in Eval() local
301 struct perm_data { struct302 int op;303 int mode;
3058 struct perm_data *perm_data; in parse_perm_args() local3118 struct perm_data *perm_data = atom->data; in perm_fn() local
207 auto& perm_data = perm_array.GetMutableBuffer<ArrayDataType::kInt32>().data; in Run() local
126 int64_t* const __restrict__ perm_data = reinterpret_cast<int64_t*>(perm_ptr + offsets[0]); in unpack_pivots_cuda_kernel() local
1115 const auto perm_data = reinterpret_cast<int64_t*>(perm_ptr); in unpack_pivots_cpu_kernel() local
4406 const int* perm_data = GetTensorData<int32_t>(&perm_tensor); in VisitTransposeNode() local
1933 let perm_data = match self.read_permission_data(r) { in set_permission_by_path() localVariable