experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improve error diagnostics in embedding_lookup.

Browse Source 
Change: 126251739
This commit is contained in:
A. Unique TensorFlower 2016-06-29 16:27:35 -08:00 committed by TensorFlower Gardener
parent 1d92cfcbf5
commit 3f488101d8
1 changed files with 2 additions and 2 deletions

4
tensorflow/python/ops/embedding_ops.py
View File

@ -73,11 +73,11 @@ def embedding_lookup(params, ids, partition_strategy="mod", name=None,
Raises: Raises:
ValueError: If `params` is empty. ValueError: If `params` is empty.
""" """
if params is None or params == []: # pylint: disable=g-explicit-bool-comparison
raise ValueError("Need at least one param")
if not isinstance(params, list): if not isinstance(params, list):
params = [params] params = [params]
with ops.op_scope(params + [ids], name, "embedding_lookup") as name: with ops.op_scope(params + [ids], name, "embedding_lookup") as name:
if not params:
raise ValueError("Need at least one param")
np = len(params) # Number of partitions np = len(params) # Number of partitions
params = ops.convert_n_to_tensor_or_indexed_slices(params, name="params") params = ops.convert_n_to_tensor_or_indexed_slices(params, name="params")
if np == 1: if np == 1: