Searched refs:kTpuAllTypes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tpu_device.cc | 461 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_TPU_NODE, XlaLocalLaunchOp, kTpuAllTypes); in RegisterTpuNodeDevice()
462 REGISTER_XLA_COMPILE_KERNEL(DEVICE_TPU_NODE, XlaCompileOp, kTpuAllTypes); in RegisterTpuNodeDevice()
463 REGISTER_XLA_RUN_KERNEL(DEVICE_TPU_NODE, XlaRunOp, kTpuAllTypes); in RegisterTpuNodeDevice()
464 REGISTER_XLA_DEVICE_KERNELS(DEVICE_TPU_NODE, kTpuAllTypes); in RegisterTpuNodeDevice()
477 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_TPU_NODE, XlaLocalLaunchOp, kTpuAllTypes);
478 REGISTER_XLA_COMPILE_KERNEL(DEVICE_TPU_NODE, XlaCompileOp, kTpuAllTypes);
479 REGISTER_XLA_RUN_KERNEL(DEVICE_TPU_NODE, XlaRunOp, kTpuAllTypes);
480 REGISTER_XLA_DEVICE_KERNELS(DEVICE_TPU_NODE, kTpuAllTypes);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_tpu_backend.cc | 24 REGISTER_XLA_BACKEND(DEVICE_TPU_XLA_JIT, kTpuAllTypes, TpuOpFilter); in RegisterTpuXlaBackend()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_defs.h | 54 static constexpr std::array<DataType, 16> kTpuAllTypes = { variable
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 3134 if (std::find(kTpuAllTypes.begin(), kTpuAllTypes.end(), in BuildExecuteNodes()
3135 handle_dtype) == kTpuAllTypes.end()) { in BuildExecuteNodes()
3142 if (std::find(kTpuAllTypes.begin(), kTpuAllTypes.end(), dtype) == in BuildExecuteNodes()
3143 kTpuAllTypes.end()) { in BuildExecuteNodes()
3198 if (std::find(kTpuAllTypes.begin(), kTpuAllTypes.end(), dtype) == in BuildExecuteNodes()
3199 kTpuAllTypes.end()) { in BuildExecuteNodes()
3215 if (std::find(kTpuAllTypes.begin(), kTpuAllTypes.end(), dtype) == in BuildExecuteNodes()
3216 kTpuAllTypes.end()) { in BuildExecuteNodes()
|