Searched refs:cost_node (Results 1 – 3 of 3) sorted by relevance
310 const CostGraphDef::Node* cost_node = it->second; in CostGraphToOpPerformanceData() local318 *perf->mutable_op()->mutable_device() = GetDeviceInfo(cost_node->device()); in CostGraphToOpPerformanceData()320 perf->set_temporary_memory_size(cost_node->temporary_memory_size()); in CostGraphToOpPerformanceData()323 perf->set_compute_cost(cost_node->compute_cost() * 1000); in CostGraphToOpPerformanceData()324 perf->set_compute_time(cost_node->compute_time() * 1000); in CostGraphToOpPerformanceData()325 perf->set_memory_time(cost_node->memory_time() * 1000); in CostGraphToOpPerformanceData()327 for (const auto& output_info : cost_node->output_info()) { in CostGraphToOpPerformanceData()332 cost_node->temporary_memory_size()); in CostGraphToOpPerformanceData()334 cost_node->persistent_memory_size()); in CostGraphToOpPerformanceData()
777 const Node* cost_node, bool is_function_op) { in AllocateNode() argument787 int cost_id = cost_node ? cost_node->cost_id() : id; in AllocateNode()
689 const Node* cost_node, bool is_function_op);