Searched refs:GetOrCalculateOutputSize (Results 1 – 2 of 2) sorted by relevance
965 device.memory_usage += GetOrCalculateOutputSize(node_state, port_num); in MarkNodeExecuted()1043 GetOrCalculateOutputSize(input_state, port); in MarkNodeExecuted()1113 output_size = GetOrCalculateOutputSize(it->second, port); in Summary()1165 op_to_memory[node->op()] += GetOrCalculateOutputSize(it->second, port); in Summary()1274 GetOrCalculateOutputSize(nodestate, slot); in GenerateRunMetadata()1355 persistent_memory_usage += GetOrCalculateOutputSize(node_state, port); in GetPersistentMemoryUsage()1368 int64_t SchedulerState::GetOrCalculateOutputSize(const NodeState& node_state, in GetOrCalculateOutputSize() function in tensorflow::grappler::SchedulerState
431 int64_t GetOrCalculateOutputSize(const NodeState& node_state,