Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h116 Device* host_cpu_device) in KernelAndDevice() argument
118 host_cpu_device_(host_cpu_device), in KernelAndDevice()
197 Device* host_cpu_device) in KernelAndDeviceOp() argument
199 host_cpu_device), in KernelAndDeviceOp()
268 Device* host_cpu_device, const string& name, in KernelAndDeviceFunc() argument
273 host_cpu_device), in KernelAndDeviceFunc()
Dexecute_node_test.cc35 Device* host_cpu_device) in TestKernelAndDeviceFunc() argument
40 host_cpu_device, /*name=*/"", /*outputs_on_op_device=*/false,
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2690 const string& host_cpu_device, in CreatePerHostDummyArgs() argument
2755 fill_def.set_device(host_cpu_device); in CreatePerHostDummyArgs()
2790 const string& host_cpu_device, int64 var_index, in CreateOrGetPerHostVariableCopy() argument
2799 auto it = per_host_var_copies->find(host_cpu_device); in CreateOrGetPerHostVariableCopy()
2806 TF_RET_CHECK(DeviceNameUtils::ParseFullName(host_cpu_device, &parsed_device)); in CreateOrGetPerHostVariableCopy()
2833 ndef.set_device(host_cpu_device); in CreateOrGetPerHostVariableCopy()
2840 id_node->set_assigned_device_name(host_cpu_device); in CreateOrGetPerHostVariableCopy()
2872 CreatePerHostDummyArgs(arg_shapes[orig_arg_num], host_cpu_device, in CreateOrGetPerHostVariableCopy()
2880 (*per_host_var_copies)[host_cpu_device] = std::move(index_mapping); in CreateOrGetPerHostVariableCopy()