Searched refs:sparse_tensor_lib (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
H A Dsparse_tensors_map_ops_test.py26 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib namespace
46 return sparse_tensor_lib.SparseTensor(
59 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
70 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
76 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
200 st = sparse_tensor_lib.SparseTensor(indices, values, shape)
H A Dsets_test.py26 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib namespace
65 return sparse_tensor_lib.SparseTensor(
127 sp = sparse_tensor_lib.SparseTensor(
314 sp_a = sparse_tensor_lib.SparseTensor(
332 sp_b = sparse_tensor_lib.SparseTensor(
458 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
799 sp_a = sparse_tensor_lib.SparseTensor(
817 sp_b = sparse_tensor_lib.SparseTensor(
1091 sp_a = sparse_tensor_lib.SparseTensor(
1109 sp_b = sparse_tensor_lib
[all...]
H A Dsparse_serialization_ops_test.py24 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib namespace
35 return sparse_tensor_lib.SparseTensor(
48 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
59 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
65 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
/external/tensorflow/tensorflow/contrib/layers/python/layers/
H A Dembedding_ops_test.py33 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib namespace
73 sparse_ids = sparse_tensor_lib.SparseTensor(
78 sparse_weights = sparse_tensor_lib.SparseTensor(
99 sparse_ids = sparse_tensor_lib.SparseTensor(
104 sparse_weights = sparse_tensor_lib.SparseTensor(
334 sparse_tensor = sparse_tensor_lib.SparseTensor(
471 sp_values = sparse_tensor_lib.SparseTensor(
485 sp_values = sparse_tensor_lib.SparseTensor(
499 sp_values = sparse_tensor_lib.SparseTensor(
524 sp_values = sparse_tensor_lib
[all...]
H A Dfeature_column_test.py33 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib namespace
56 return sparse_tensor_lib.SparseTensor(
169 input_tensor_c1 = sparse_tensor_lib.SparseTensor(
172 input_tensor_c2 = sparse_tensor_lib.SparseTensor(
567 sparse_tensor = sparse_tensor_lib.SparseTensor(
1016 sparse_tensor_lib.SparseTensor))
1019 sparse_tensor_lib.SparseTensor))
1041 input_tensor = sparse_tensor_lib.SparseTensor(
1100 input_tensor = sparse_tensor_lib.SparseTensor(
/external/tensorflow/tensorflow/contrib/estimator/python/estimator/
H A Dextenders.py27 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib namespace
228 feature = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
/external/tensorflow/tensorflow/python/keras/_impl/keras/
H A Destimator.py30 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib namespace
130 sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(labels))
/external/tensorflow/tensorflow/python/feature_column/
H A Dfeature_column.py140 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib namespace
1911 feature_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
1916 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
1973 input_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
1975 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
1993 return sparse_tensor_lib.SparseTensor(
2052 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
2147 sparse_tensor = sparse_tensor_lib.SparseTensor(
2428 if not isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
2448 return sparse_tensor_lib
[all...]
/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py36 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib namespace
926 sparse_tensor_lib.SparseTensor.from_value(t)
927 if sparse_tensor_lib.is_sparse(t) else ops.convert_to_tensor(
966 sparse_tensor_lib.SparseTensor.from_value(t)
967 if sparse_tensor_lib.is_sparse(t) else ops.convert_to_tensor(
1008 if not isinstance(sparse_tensor, sparse_tensor_lib.SparseTensor):
1579 sparse_tensor_lib.SparseTensor.from_value(t)
1580 if sparse_tensor_lib.is_sparse(t) else t for t in nest.flatten(ret)

Completed in 315 milliseconds