Home
last modified time | relevance | path

Searched refs:SetNumOutputs (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.h83 void SetNumOutputs(const Node* node, int num_outputs);
Dcostmodel.cc153 void CostModel::SetNumOutputs(const Node* node, int num_outputs) { in SetNumOutputs() function in tensorflow::CostModel
416 cost_model->SetNumOutputs(n, num_outputs); in AddNodesToCostModel()