Home
last modified time | relevance | path

Searched refs:host_compute (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc797 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/
Ddfs_hlo_visitor_with_default.h106 Status HandleHostCompute(HloInstructionPtr host_compute) override { in HandleHostCompute() argument
107 return DefaultAction(host_compute); in HandleHostCompute()
Dhlo_cost_analysis.h74 Status HandleHostCompute(const HloInstruction* host_compute) override;