Searched refs:MaxMemoryShape (Results 1 – 3 of 3) sorted by relevance
134 const TensorShapeProto& MaxMemoryShape(const Node* node,
286 const TensorShapeProto& CostModel::MaxMemoryShape(const Node* node, in MaxMemoryShape() function in tensorflow::CostModel525 *output_info->mutable_shape() = MaxMemoryShape(n, i); in AddToCostGraphDef()
106 TensorShapeProto shape = cm->MaxMemoryShape(node, 0); in TEST()224 TensorShapeProto shape = cm->MaxMemoryShape(node, 0); in TestHWAccelerator()