Searched refs:MaxMemorySize (Results 1 – 3 of 3) sorted by relevance
122 Bytes MaxMemorySize(const Node* node, int output_slot) const;
275 Bytes CostModel::MaxMemorySize(const Node* node, int slot) const { in MaxMemorySize() function in tensorflow::CostModel526 output_info->set_size(MaxMemorySize(n, i).value()); in AddToCostGraphDef()
105 EXPECT_LE(8, cm->MaxMemorySize(node, 0)); in TEST()232 EXPECT_LE(8, cm->MaxMemorySize(node, 0)); in TestHWAccelerator()