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
617 const auto& axis_tensor = context->tensors[node->inputs->data[2]]; in ComputeSplitVUnknownSplitSize() local2328 const auto& axis_tensor = context->tensors[node->inputs->data[1]]; in Validate() local5894 const TfLiteTensor& axis_tensor = context->tensors[axis_id]; in AddOpsAndTensors() local5978 const TfLiteTensor& axis_tensor = in AddOpsAndTensors() local
781 const TfLiteTensor* axis_tensor = reader->GetInputTensor(1); in Parse() local2275 const TfLiteTensor* axis_tensor = reader->GetInputTensor(0); in Parse() local2317 const TfLiteTensor* axis_tensor = reader->GetInputTensor(2); in Parse() local
1021 const Tensor* axis_tensor = c->input_tensor(1); in __anon38bbb0e81402() local
3832 Tensor axis_tensor; in GetConcatAxis() local
4214 Tensor axis_tensor; in IsAxis0() local