Searched refs:TpuProgramShardingType (Results 1 – 4 of 4) sorted by relevance
120 tpu_program_group.tpu_programs(TpuProgramShardingType::kSharding)); in InitializeEntry()128 tpu_program_group.tpu_programs(TpuProgramShardingType::kUnsharding)); in InitializeEntry()
129 std::vector<XLA_TpuProgram*> tpu_programs(TpuProgramShardingType type) const;
316 TpuProgramShardingType sharding_type) const { in tpu_programs()
37 enum TpuProgramShardingType { kInvalid = 0, kMain, kSharding, kUnsharding }; enum369 XLA_TpuProgram* tpu_program, TpuProgramShardingType type);