Searched defs:axis_tensor (Results 1 – 9 of 9) sorted by relevance
57 const TfLiteTensor* axis_tensor = GetInput(context, node, kAxisTensor); in Eval() local
77 const TfLiteTensor* axis_tensor; in Eval() local
67 const Tensor& axis_tensor = c->input(2); in Compute() local
100 const Tensor& axis_tensor = context->input(1); in Compute() local
1711 const auto& axis_tensor = context->tensors[node->inputs->data[1]]; in Validate() local4496 const TfLiteTensor& axis_tensor = context->tensors[axis_id]; in AddOpsAndTensors() local4579 const TfLiteTensor& axis_tensor = in AddOpsAndTensors() local
1003 const Tensor* axis_tensor = c->input_tensor(1); in __anondb9326b21402() local
1840 const TfLiteTensor* axis_tensor = reader->GetInputTensor(2); in Parse() local
3773 Tensor axis_tensor; in GetConcatAxis() local
3589 Tensor axis_tensor; in IsAxis0() local