Searched refs:TempMemorySize (Results 1 – 2 of 2) sorted by relevance
134 Bytes TempMemorySize(const Node* node) const;
304 Bytes CostModel::TempMemorySize(const Node* node) const { in TempMemorySize() function in tensorflow::CostModel534 cnode->set_temporary_memory_size(TempMemorySize(n).value()); in AddToCostGraphDef()