Home
last modified time | relevance | path

Searched refs:compile_device (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h362 int num_cores_per_replica, const string& compile_device,
Ddistributed_tpu_rewrite_pass.cc2315 int num_cores_per_replica, const string& compile_device, in BuildCompileNode() argument
2428 def.set_device(compile_device); in BuildCompileNode()
2444 (*compile_node)->set_assigned_device_name(compile_device); in BuildCompileNode()