Searched refs:tpu_program_group_interface (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
H A D | BUILD | 70 ":tpu_program_group_interface", 324 ":tpu_program_group_interface", 387 ":tpu_program_group_interface", 394 name = "tpu_program_group_interface", 395 hdrs = ["tpu_program_group_interface.h"], 418 ":tpu_program_group_interface", 539 ":tpu_program_group_interface", 604 ":tpu_program_group_interface", 642 ":tpu_program_group_interface", 748 ":tpu_program_group_interface",
|
H A D | tpu_program_group.cc | 263 TpuProgramGroupInterface* tpu_program_group_interface) { in CompileAndBuild() argument 286 tensorflow::down_cast<TpuProgramGroup*>(tpu_program_group_interface); in CompileAndBuild() 297 TpuProgramGroupInterface* tpu_program_group_interface) { in CompileAndBuild() argument 322 tensorflow::down_cast<TpuProgramGroup*>(tpu_program_group_interface); in CompileAndBuild()
|
H A D | tpu_program_group.h | 96 TpuProgramGroupInterface* tpu_program_group_interface); 102 TpuProgramGroupInterface* tpu_program_group_interface);
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
H A D | BUILD | 96 "//tensorflow/core/tpu/kernels:tpu_program_group_interface",
|