Searched refs:kInputShardingAttr (Results 1 – 4 of 4) sorted by relevance
32 extern const char* const kInputShardingAttr;
40 const char* const kInputShardingAttr = "input_sharding_configuration"; variable223 kInputShardingAttr); in ExtractInputsForLogicalDevices()
222 op->getAttrOfType<ArrayAttr>(tensorflow::kInputShardingAttr); in SetMetadataProtoArgs()225 CreateMissingAttributeMsg(tensorflow::kInputShardingAttr)); in SetMetadataProtoArgs()229 llvm::formatv(kBadArrayAttrLengthMsg, tensorflow::kInputShardingAttr, in SetMetadataProtoArgs()263 tensorflow::kInputShardingAttr, index, in SetMetadataProtoArgs()
182 cluster_func->setAttr(tensorflow::kInputShardingAttr, in IdentifyXlaShardingForComputationInputs()