Searched refs:kSharding (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compilation_cache_external.cc | 120 tpu_program_group.tpu_programs(TpuProgramShardingType::kSharding)); in InitializeEntry()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_ops_c_api.h | 37 enum TpuProgramShardingType { kInvalid = 0, kMain, kSharding, kUnsharding }; enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser.cc | 322 kSharding, enumerator
1001 attrs["sharding"] = {/*required=*/false, AttrTy::kSharding, &sharding}; in ParseInstructionRhs()
2905 attrs["entry"] = {/*required=*/true, AttrTy::kSharding, &entry_sharding}; in ParseDomain()
2906 attrs["exit"] = {/*required=*/true, AttrTy::kSharding, &exit_sharding}; in ParseDomain()
3684 case AttrTy::kSharding: { in ParseAttributeHelper()
|
| /external/libabigail/tests/data/test-read-dwarf/ |
| D | test11-pr18828.so.abi | 15774 <enumerator name='kSharding' value='10'/>
|
| D | test12-pr18844.so.abi | 18973 <enumerator name='kSharding' value='10'/>
|