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