Searched refs:DEVICE_TPU_SYSTEM (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | control_flow_ops.cc | 58 Name("Switch").Device(DEVICE_TPU_SYSTEM).HostMemory("pred"), SwitchOp);
240 Name("Merge").Device(DEVICE_TPU_SYSTEM).HostMemory("value_index"), MergeOp);
304 REGISTER_KERNEL_BUILDER(Name("Enter").Device(DEVICE_TPU_SYSTEM), EnterOp);
361 REGISTER_KERNEL_BUILDER(Name("Exit").Device(DEVICE_TPU_SYSTEM), ExitOp);
414 REGISTER_KERNEL_BUILDER(Name("NextIteration").Device(DEVICE_TPU_SYSTEM),
480 .Device(DEVICE_TPU_SYSTEM)
488 REGISTER_KERNEL_BUILDER(Name("ControlTrigger").Device(DEVICE_TPU_SYSTEM),
|
| D | sendrecv_ops.cc | 127 REGISTER_KERNEL_BUILDER(Name("_Send").Device(DEVICE_TPU_SYSTEM), SendOp);
128 REGISTER_KERNEL_BUILDER(Name("_HostSend").Device(DEVICE_TPU_SYSTEM), SendOp);
221 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_TPU_SYSTEM), RecvOp);
222 REGISTER_KERNEL_BUILDER(Name("_HostRecv").Device(DEVICE_TPU_SYSTEM), RecvOp);
|
| D | no_op.cc | 21 REGISTER_KERNEL_BUILDER(Name("NoOp").Device(DEVICE_TPU_SYSTEM), NoOp);
|
| D | identity_n_op.cc | 28 REGISTER_KERNEL_BUILDER(Name("IdentityN").Device(DEVICE_TPU_SYSTEM),
|
| D | identity_op.cc | 27 REGISTER_KERNEL_BUILDER(Name("Identity").Device(DEVICE_TPU_SYSTEM), IdentityOp);
|
| D | constant_op.cc | 97 REGISTER_KERNEL_BUILDER(Name("Const").Device(DEVICE_TPU_SYSTEM), ConstantOp);
|
| D | function_ops.cc | 95 REGISTER_KERNEL_BUILDER(Name(kRetOp).Device(DEVICE_TPU_SYSTEM), RetvalOp);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_configuration_ops.cc | 408 .Device(DEVICE_TPU_SYSTEM)
412 .Device(DEVICE_TPU_SYSTEM)
417 Name("_ShutdownDistributedTPU").Device(DEVICE_TPU_SYSTEM),
420 .Device(DEVICE_TPU_SYSTEM)
425 Name("_SetGlobalTPUArray").Device(DEVICE_TPU_SYSTEM).HostMemory("topology"),
428 .Device(DEVICE_TPU_SYSTEM)
|
| D | replication_ops.cc | 24 REGISTER_KERNEL_BUILDER(Name("_TPUReplicate").Device(DEVICE_TPU_SYSTEM),
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_helpers.cc | 39 if (!system_spec->has_type || system_spec->type != DEVICE_TPU_SYSTEM) { in GetSystemDevice()
40 system_spec->type = DEVICE_TPU_SYSTEM; in GetSystemDevice()
104 strings::StrCat("/job:", system_spec.job, "/device:", DEVICE_TPU_SYSTEM, in GetHostSystemDevices()
112 strings::StrCat("/device:", DEVICE_TPU_SYSTEM, ":0"), &host_spec)); in GetHostSystemDevices()
|
| D | distributed_tpu_rewrite_pass.cc | 1372 str_util::StringReplace(replication_device->name(), DEVICE_TPU_SYSTEM, in GetTPUDeviceNames()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tpu_device.cc | 434 absl::StrCat(name_prefix, "/device:", DEVICE_TPU_SYSTEM, ":", 0), in CreateDevices()
435 DeviceType(DEVICE_TPU_SYSTEM), Bytes(memory_limit), DeviceLocality(), in CreateDevices()
436 absl::StrCat("device: ", DEVICE_TPU_SYSTEM, " device")); in CreateDevices()
469 REGISTER_LOCAL_DEVICE_FACTORY(DEVICE_TPU_SYSTEM, TpuSystemDeviceFactory); in RegisterTpuSystemDevice()
482 REGISTER_LOCAL_DEVICE_FACTORY(DEVICE_TPU_SYSTEM, TpuSystemDeviceFactory);
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | types.cc | 41 const char* const DEVICE_TPU_SYSTEM = "TPU_SYSTEM"; variable
|
| D | types.h | 77 TF_EXPORT extern const char* const DEVICE_TPU_SYSTEM; // "TPU_SYSTEM"
|