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