Searched refs:host_compute (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/ |
D | encapsulate_subgraphs_pass.cc | 797 Node* host_compute = graph_->AddNode(host_compute_def, &s); in AddHostComputes() local 799 oc_subgraph.host_compute_name = host_compute->name(); in AddHostComputes() 807 graph_->AddEdge(src_image, src_slot, host_compute, input_index); in AddHostComputes() 814 graph_->AddControlEdge(src_image, host_compute); in AddHostComputes() 824 graph_->AddEdge(host_compute, output_index, dst_image, dst_slot); in AddHostComputes() 831 graph_->AddControlEdge(host_compute, dst_image); in AddHostComputes() 932 Node* host_compute = nullptr; in AddShapeInferenceInfo() local 935 host_compute = n; in AddShapeInferenceInfo() 939 if (host_compute == nullptr) { in AddShapeInferenceInfo() 947 host_compute->AddAttr("shape_inference_graph", ""); in AddShapeInferenceInfo() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | dfs_hlo_visitor_with_default.h | 106 Status HandleHostCompute(HloInstructionPtr host_compute) override { in HandleHostCompute() argument 107 return DefaultAction(host_compute); in HandleHostCompute()
|
D | hlo_cost_analysis.h | 74 Status HandleHostCompute(const HloInstruction* host_compute) override;
|