Searched refs:axis_const (Results 1 – 1 of 1) sorted by relevance
38 auto* axis_const = graph_builder_->AddConstNodeWithData( in PopulateSubGraph() local40 AddInput(TensorID(axis_const->GetID(), 0)); in PopulateSubGraph()