Searched defs:proto_arg (Results 1 – 2 of 2) sorted by relevance
91 const tpu::TPUCompileMetadataProto::Arg& proto_arg, const int arg_index, in SetPerCoreArgShapes()189 const tpu::TPUCompileMetadataProto::Arg& proto_arg = metadata_.args(i); in BuildComputationArgumentDescriptions() local262 const auto& proto_arg = metadata_.args(i); in GetShardingInfo() local
210 const tpu::TPUCompileMetadataProto::Arg& proto_arg = metadata.args(i); in AddVariableUpdatesToCores() local