Searched refs:tpu_mesh_common_state (Results 1 – 2 of 2) sorted by relevance
230 tensorflow::TpuMeshCommonState* tpu_mesh_common_state; member
218 params.tpu_mesh_common_state = mesh_common_state; in Compute()