Searched defs:key_tensor_proto (Results 1 – 2 of 2) sorted by relevance
770 auto *key_tensor_proto = map_tensor_proto->mutable_key_tensor(); in ConvertMapParameterToMapTensorProto() local811 auto *key_tensor_proto = attr_proto->add_tensors(); in ConvertAbstractMapTensorToAttrProto() local
799 const auto &key_tensor_proto = attr_proto.tensors(0); in BuildAbstractMapTensorFromAttrProto() local893 const auto &key_tensor_proto = map_parameter_proto.key_tensor(); in BuildMapParameterFromMapTensorProto() local