| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_gather_decomposer.cc | 95 auto shard_id = comp->AddInstruction(HloInstruction::CreateDynamicSlice( in DecomposeAllGather() local
97 shard_id = comp->AddInstruction( in DecomposeAllGather()
98 HloInstruction::CreateReshape(ShapeUtil::MakeShape(U32, {}), shard_id)); in DecomposeAllGather()
99 return shard_id; in DecomposeAllGather()
101 HloInstruction* shard_id; in DecomposeAllGather() local
112 shard_id = shard_id_from_subgroup(global_id); in DecomposeAllGather()
116 shard_id = comp->AddInstruction(HloInstruction::CreatePartitionId()); in DecomposeAllGather()
119 shard_id = shard_id_from_subgroup( in DecomposeAllGather()
124 shard_id->shape(), HloOpcode::kMultiply, shard_id, in DecomposeAllGather()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | naming.cc | 28 string DataFilename(StringPiece prefix, int32 shard_id, int32 num_shards) { in DataFilename() argument
30 DCHECK_LT(shard_id, num_shards); in DataFilename()
33 shard_id, num_shards); in DataFilename()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LoadTPUEmbeddingStochasticGradientDescentParameters.pbtxt | 28 name: "shard_id"
60 name: "shard_id"
97 name: "shard_id"
|
| D | RetrieveTPUEmbeddingStochasticGradientDescentParameters.pbtxt | 28 name: "shard_id"
60 name: "shard_id"
97 name: "shard_id"
|
| D | LoadTPUEmbeddingAdagradParameters.pbtxt | 32 name: "shard_id"
68 name: "shard_id"
109 name: "shard_id"
|
| D | RetrieveTPUEmbeddingProximalAdagradParameters.pbtxt | 32 name: "shard_id"
68 name: "shard_id"
109 name: "shard_id"
|
| D | RetrieveTPUEmbeddingMomentumParameters.pbtxt | 32 name: "shard_id"
68 name: "shard_id"
109 name: "shard_id"
|
| D | RetrieveTPUEmbeddingAdagradParameters.pbtxt | 32 name: "shard_id"
68 name: "shard_id"
109 name: "shard_id"
|
| D | LoadTPUEmbeddingMomentumParameters.pbtxt | 32 name: "shard_id"
68 name: "shard_id"
109 name: "shard_id"
|
| D | LoadTPUEmbeddingProximalAdagradParameters.pbtxt | 32 name: "shard_id"
68 name: "shard_id"
109 name: "shard_id"
|
| D | LoadTPUEmbeddingRMSPropParameters.pbtxt | 36 name: "shard_id"
76 name: "shard_id"
121 name: "shard_id"
|
| D | RetrieveTPUEmbeddingRMSPropParameters.pbtxt | 36 name: "shard_id"
76 name: "shard_id"
121 name: "shard_id"
|
| D | LoadTPUEmbeddingAdadeltaParameters.pbtxt | 36 name: "shard_id"
76 name: "shard_id"
121 name: "shard_id"
|
| D | LoadTPUEmbeddingFTRLParameters.pbtxt | 36 name: "shard_id"
76 name: "shard_id"
121 name: "shard_id"
|
| D | LoadTPUEmbeddingMDLAdagradLightParameters.pbtxt | 40 name: "shard_id"
84 name: "shard_id"
133 name: "shard_id"
|
| D | LoadTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt | 36 name: "shard_id"
76 name: "shard_id"
121 name: "shard_id"
|
| D | RetrieveTPUEmbeddingFTRLParameters.pbtxt | 36 name: "shard_id"
76 name: "shard_id"
121 name: "shard_id"
|
| D | LoadTPUEmbeddingCenteredRMSPropParameters.pbtxt | 40 name: "shard_id"
84 name: "shard_id"
133 name: "shard_id"
|
| D | RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt | 36 name: "shard_id"
76 name: "shard_id"
121 name: "shard_id"
|
| D | LoadTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt | 36 name: "shard_id"
76 name: "shard_id"
121 name: "shard_id"
|
| D | RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.pbtxt | 40 name: "shard_id"
84 name: "shard_id"
133 name: "shard_id"
|
| D | RetrieveTPUEmbeddingADAMParametersGradAccumDebug.pbtxt | 40 name: "shard_id"
84 name: "shard_id"
133 name: "shard_id"
|
| D | RetrieveTPUEmbeddingCenteredRMSPropParameters.pbtxt | 40 name: "shard_id"
84 name: "shard_id"
133 name: "shard_id"
|
| D | LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.pbtxt | 40 name: "shard_id"
84 name: "shard_id"
133 name: "shard_id"
|
| D | RetrieveTPUEmbeddingMDLAdagradLightParameters.pbtxt | 40 name: "shard_id"
84 name: "shard_id"
133 name: "shard_id"
|