Home
last modified time | relevance | path

Searched defs:perm_data (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dtranspose.cc51 const int* perm_data = GetTensorData<int32_t>(op_context->perm); in ResizeOutputTensor() local
99 const int* perm_data = GetTensorData<int32_t>(op_context.perm); in Eval() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunpartition_embedding_lookup.cc207 auto& perm_data = perm_array.GetMutableBuffer<ArrayDataType::kInt32>().data; in Run() local
/external/pytorch/aten/src/ATen/native/cuda/
DLinearAlgebra.cu126 int64_t* const __restrict__ perm_data = reinterpret_cast<int64_t*>(perm_ptr + offsets[0]); in unpack_pivots_cuda_kernel() local
/external/squashfs-tools/squashfs-tools/
Daction.h301 struct perm_data { struct
302 int op;
303 int mode;
Daction.c3058 struct perm_data *perm_data; in parse_perm_args() local
3118 struct perm_data *perm_data = atom->data; in perm_fn() local
/external/pytorch/aten/src/ATen/native/
DBatchLinearAlgebraKernel.cpp1115 const auto perm_data = reinterpret_cast<int64_t*>(perm_ptr); in unpack_pivots_cpu_kernel() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc4406 const int* perm_data = GetTensorData<int32_t>(&perm_tensor); in VisitTransposeNode() local
/external/crosvm/devices/src/virtio/fs/
Dpassthrough.rs1940 let perm_data = match self.read_permission_data(r) { in set_permission_by_path() localVariable