Home
last modified time | relevance | path

Searched refs:MaxMemorySize (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.h130 Bytes MaxMemorySize(const Node* node, int output_slot) const;
Dcostmodel.cc277 Bytes CostModel::MaxMemorySize(const Node* node, int slot) const { in MaxMemorySize() function in tensorflow::CostModel
529 output_info->set_size(MaxMemorySize(n, i).value()); in AddToCostGraphDef()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc105 EXPECT_LE(8, cm->MaxMemorySize(node, 0)); in TEST()
223 EXPECT_LE(8, cm->MaxMemorySize(node, 0)); in TestHWAccelerator()