Searched refs:perm_data (Results 1 – 5 of 5) sorted by relevance
51 const int* perm_data = GetTensorData<int32_t>(op_context->perm); in ResizeOutputTensor() local57 TF_LITE_ENSURE_MSG(context, (perm_data[idx] >= 0 && perm_data[idx] < dims), in ResizeOutputTensor()65 output_size->data[idx] = input_size->data[perm_data[idx]]; in ResizeOutputTensor()99 const int* perm_data = GetTensorData<int32_t>(op_context.perm); in Eval() local104 params.perm[i] = perm_data[i]; in Eval()
207 auto& perm_data = perm_array.GetMutableBuffer<ArrayDataType::kInt32>().data; in Run() local208 perm_data.resize(RequiredBufferSizeForShape(perm_array.shape())); in Run()212 perm_data[i] = p * partition_array_dims[0] + i / num_partitions; in Run()
3058 struct perm_data *perm_data; in parse_perm_args() local3096 perm_data = malloc(sizeof(struct perm_data)); in parse_perm_args()3097 if (perm_data == NULL) in parse_perm_args()3100 perm_data->op = op; in parse_perm_args()3101 perm_data->mode = mode; in parse_perm_args()3103 atom->data = perm_data; in parse_perm_args()3118 struct perm_data *perm_data = atom->data; in perm_fn() local3121 switch(perm_data->op) { in perm_fn()3123 return (buf->st_mode & ~S_IFMT) == perm_data->mode; in perm_fn()3125 return (buf->st_mode & perm_data->mode) == perm_data->mode; in perm_fn()[all …]
301 struct perm_data { struct