Home
last modified time | relevance | path

Searched defs:manual_sharding (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dsharding_propagation.cc926 HloSharding* auto_sharding, HloSharding* manual_sharding) { in RefineManualAutoShardingFromAuto()
998 HloSharding* auto_sharding, HloSharding* manual_sharding) { in RefineManualAutoShardingFromManual()
1077 HloSharding manual_sharding = man_conversion_op->sharding(); in InferUnspecifiedDimsFromOperand() local
1088 HloSharding manual_sharding = annotate_op->sharding(); in InferUnspecifiedDimsFromOperand() local
1130 HloSharding manual_sharding = man_conversion_op->sharding(); in InferUnspecifiedDimsFromOneUser() local
1140 HloSharding manual_sharding = annotate_op->sharding(); in InferUnspecifiedDimsFromOneUser() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
H A Dxla_sharding.py449 manual_sharding, argument
477 manual_sharding, argument
/aosp_15_r20/external/googleapis/google/devtools/testing/v1/
H A Dtest_execution.proto1270 ManualSharding manual_sharding = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc5062 const OpSharding& manual_sharding, in ConvertSpmdFullToShardShape()
5107 const OpSharding& manual_sharding, in ConvertSpmdShardToFullShape()