Home
last modified time | relevance | path

Searched refs:sparse_tensor_lib (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_tensors_map_ops_test.py22 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib unknown
44 return sparse_tensor_lib.SparseTensor(
57 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
68 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
74 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
204 st = sparse_tensor_lib.SparseTensor(indices, values, shape)
H A Dsparse_serialization_ops_test.py21 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib unknown
34 return sparse_tensor_lib.SparseTensor(
47 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
58 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
64 return sparse_tensor_lib.SparseTensorValue(ind, val, shape)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
H A Dsets_test.py23 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib unknown
63 return sparse_tensor_lib.SparseTensor(
128 sp = sparse_tensor_lib.SparseTensor(
319 sp_a = sparse_tensor_lib.SparseTensor(
337 sp_b = sparse_tensor_lib.SparseTensor(
463 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
809 sp_a = sparse_tensor_lib.SparseTensor(
827 sp_b = sparse_tensor_lib.SparseTensor(
1105 sp_a = sparse_tensor_lib.SparseTensor(
1123 sp_b = sparse_tensor_lib.SparseTensor(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/
H A Dfeature_column.py141 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib unknown
2213 feature_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
2218 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
2275 input_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
2277 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
2295 return sparse_tensor_lib.SparseTensor(
2368 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
2465 sparse_tensor = sparse_tensor_lib.SparseTensor(
2781 if not isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
2801 return sparse_tensor_lib.SparseTensor(
[all …]
H A Dfeature_column_v2.py141 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib unknown
2481 feature_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
2486 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
2530 input_tensor = sparse_tensor_lib.convert_to_tensor_or_sparse_tensor(
2532 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
2550 return sparse_tensor_lib.SparseTensor(
2635 if isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
2840 sparse_tensor = sparse_tensor_lib.SparseTensor(
3397 if not isinstance(input_tensor, sparse_tensor_lib.SparseTensor):
3417 return sparse_tensor_lib.SparseTensor(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py46 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib unknown
4749 if not isinstance(sparse_tensor, sparse_tensor_lib.SparseTensor):