Searched refs:perm_data (Results 1 – 6 of 6) sorted by relevance
48 const int* perm_data = GetTensorData<int32_t>(op_context->perm); in ResizeOutputTensor() local54 TF_LITE_ENSURE_MSG(context, (perm_data[idx] >= 0 && perm_data[idx] < dims), in ResizeOutputTensor()62 output_size->data[idx] = input_size->data[perm_data[idx]]; in ResizeOutputTensor()95 const int* perm_data = GetTensorData<int32_t>(op_context.perm); in Eval() local100 params.perm[i] = perm_data[i]; in Eval()
205 auto& perm_data = perm_array.GetMutableBuffer<ArrayDataType::kInt32>().data; in Run() local206 perm_data.resize(RequiredBufferSizeForShape(perm_array.shape())); in Run()210 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
3933 static const uint8 perm_data[64] = {3954 const int8x16_t perm_data_0 = vld1q_u8(perm_data);3955 const int8x16_t perm_data_1 = vld1q_u8(perm_data + 16);3956 const int8x16_t perm_data_2 = vld1q_u8(perm_data + 32);3957 const int8x16_t perm_data_3 = vld1q_u8(perm_data + 48);
1216 static const uint8 perm_data[64] = {1237 const int8x16_t perm_data_0 = vld1q_u8(perm_data);1238 const int8x16_t perm_data_1 = vld1q_u8(perm_data + 16);1239 const int8x16_t perm_data_2 = vld1q_u8(perm_data + 32);1240 const int8x16_t perm_data_3 = vld1q_u8(perm_data + 48);