Searched refs:kOutputShardingAttr (Results 1 – 4 of 4) sorted by relevance
33 extern const char* const kOutputShardingAttr;
41 const char* const kOutputShardingAttr = "output_sharding_configuration"; variable319 kOutputShardingAttr); in ParseAndValidateOutputSharding()
277 op->getAttrOfType<ArrayAttr>(tensorflow::kOutputShardingAttr); in SetMetadataProtoRetvals()280 CreateMissingAttributeMsg(tensorflow::kOutputShardingAttr)); in SetMetadataProtoRetvals()284 llvm::formatv(kBadArrayAttrLengthMsg, tensorflow::kOutputShardingAttr, in SetMetadataProtoRetvals()290 tensorflow::kOutputShardingAttr, in SetMetadataProtoRetvals()
298 cluster_func->setAttr(tensorflow::kOutputShardingAttr, in IdentifyXlaShardingForComputationOutputs()