Searched refs:InferMemUsageForNeighbors (Results 1 – 2 of 2) sorted by relevance
82 int64 node_memory_usage = InferMemUsageForNeighbors(outputs); in InferMemUsageForNodes()92 node_memory_usage += InferMemUsageForNeighbors(inputs); in InferMemUsageForNodes()99 int64 GraphMemory::InferMemUsageForNeighbors( in InferMemUsageForNeighbors() function in tensorflow::grappler::GraphMemory
67 int64 InferMemUsageForNeighbors(