Searched defs:axes_tensor (Results 1 – 3 of 3) sorted by relevance
41 const auto& axes_tensor = context->tensors[axes_tensor_id]; in PopulateSubGraph() local
2512 const Tensor* axes_tensor = c->input_tensor(3); in SparseReduceShapeFn() local
3413 const TfLiteTensor& axes_tensor = tensors[node->inputs->data[1]]; in VisitMeanNode() local