Searched defs:proto_arg (Results 1 – 4 of 4) sorted by relevance
52 ProtoArg proto_arg; in SerializeProto() local
49 const tpu::TPUCompileMetadataProto::Arg& proto_arg, const int arg_index, in SetPerCoreArgShapes()269 const tpu::TPUCompileMetadataProto::Arg& proto_arg = metadata.args(i); in BuildComputationArgumentDescriptions() local465 const auto& proto_arg = metadata.args(i); in GetShardingInfo() local
156 ProtoArg proto_arg; in GetDeserializedProto() local
212 const tpu::TPUCompileMetadataProto::Arg& proto_arg = metadata.args(i); in AddVariableUpdatesToCores() local