Searched refs:kTpuMeshStateInterfaceResourceName (Results 1 – 4 of 4) sorted by relevance
45 tpu::kTpuMeshStateInterfaceResourceName, state) in GetTpuMeshStateInterface()131 rmgr, tpu::kTpuMeshStateInterfaceResourceName)); in Compute()136 tpu::kTpuMeshStateInterfaceResourceName, tpu_mesh)); in Compute()241 rmgr, tpu::kTpuMeshStateInterfaceResourceName)); in Compute()263 rmgr, tpu::kTpuMeshStateInterfaceResourceName)); in Compute()266 tpu::kTpuMeshStateInterfaceResourceName, in Compute()
31 const char kTpuMeshStateInterfaceResourceName[] = "tpu_mesh_common_state"; variable
637 kTpuMeshStateInterfaceResourceName, in ComputeInternal()
161 tensorflow::tpu::kTpuMeshStateInterfaceResourceName, in Compute()