Searched refs:cost_node (Results 1 – 3 of 3) sorted by relevance
309 const CostGraphDef::Node* cost_node = it->second; in CostGraphToOpPerformanceData() local317 *perf->mutable_op()->mutable_device() = GetDeviceInfo(cost_node->device()); in CostGraphToOpPerformanceData()319 perf->set_temporary_memory_size(cost_node->temporary_memory_size()); in CostGraphToOpPerformanceData()322 perf->set_compute_cost(cost_node->compute_cost() * 1000); in CostGraphToOpPerformanceData()323 perf->set_compute_time(cost_node->compute_time() * 1000); in CostGraphToOpPerformanceData()324 perf->set_memory_time(cost_node->memory_time() * 1000); in CostGraphToOpPerformanceData()326 for (const auto& output_info : cost_node->output_info()) { in CostGraphToOpPerformanceData()331 cost_node->temporary_memory_size()); in CostGraphToOpPerformanceData()333 cost_node->persistent_memory_size()); in CostGraphToOpPerformanceData()
745 const Node* cost_node) { in AllocateNode() argument755 int cost_id = cost_node ? cost_node->cost_id() : id; in AllocateNode()
663 const Node* cost_node);