Searched refs:axis_tensor_id (Results 1 – 2 of 2) sorted by relevance
39 const int axis_tensor_id = inputs->data[1]; in PopulateSubGraph() local40 const auto& axis = context->tensors[axis_tensor_id]; in PopulateSubGraph()
35 const int axis_tensor_id = inputs->data[0]; in PopulateSubGraph() local36 const auto& axis = context->tensors[axis_tensor_id]; in PopulateSubGraph()