Searched refs:tensor_dict (Results 1 – 4 of 4) sorted by relevance
305 def _construct_sparse_tensors_for_sparse_features(features, tensor_dict): argument326 tensor_dict = dict(tensor_dict) # Do not modify argument passed in.332 sp_ids = tensor_dict[feature.index_key]334 sp_ids = [tensor_dict[index_key] for index_key in feature.index_key]335 sp_values = tensor_dict[feature.value_key]336 tensor_dict[key] = sparse_ops.sparse_merge(343 for key in set(tensor_dict) - set(features):344 del tensor_dict[key]345 return tensor_dict
975 tensor_dict = {}978 tensor_dict[str(id(x))] = y992 str(id(tensor)) in tensor_dict997 lambda t: tensor_dict[str(id(t))], node.input_tensors)1015 tensor_dict[str(id(x))] = y1020 assert str(id(x)) in tensor_dict, 'Could not compute output ' + str(x)1021 tensor = tensor_dict[str(id(x))]
247 def _tensor_dict_to_tensorinfo(tensor_dict): argument249 for key, value in tensor_dict.items()}
3729 def _verify_cross_hosts_transfer_size(tensor_dict, message): argument3732 for key, tensor in tensor_dict.items():