Searched refs:ret_map (Results 1 – 2 of 2) sorted by relevance
25 Status GetInputMap(OpKernelContext* ctx, int index, const TensorMap** ret_map) { in GetInputMap() argument36 *ret_map = map; in GetInputMap()
339 const ::tensorflow::protobuf::Map<string, string>& ret_map, in AddReturnNode() argument341 auto ret_iter = ret_map.find(ret_def.name()); in AddReturnNode()342 if (ret_iter == ret_map.end()) { in AddReturnNode()