Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.h134 const TensorShapeProto& MaxMemoryShape(const Node* node,
Dcostmodel.cc286 const TensorShapeProto& CostModel::MaxMemoryShape(const Node* node, in MaxMemoryShape() function in tensorflow::CostModel
525 *output_info->mutable_shape() = MaxMemoryShape(n, i); in AddToCostGraphDef()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc106 TensorShapeProto shape = cm->MaxMemoryShape(node, 0); in TEST()
224 TensorShapeProto shape = cm->MaxMemoryShape(node, 0); in TestHWAccelerator()