Lines Matching refs:sparse_tensor
25 from tensorflow.python.framework import sparse_tensor
47 return sparse_tensor.SparseTensor(
57 return sparse_tensor.SparseTensor(
68 return sparse_tensor.SparseTensor(
121 indices = sparse_tensor.SparseTensorValue(
124 values = sparse_tensor.SparseTensorValue(
131 return (sparse_tensor.SparseTensor.from_value(indices),
132 sparse_tensor.SparseTensor.from_value(values))
151 sparse_tensor.SparseTensor.from_value(indices_v)):
153 sparse_tensor.SparseTensor.from_value(values_v)):
220 indices0 = sparse_tensor.SparseTensorValue(
223 indices1 = sparse_tensor.SparseTensorValue(
226 values = sparse_tensor.SparseTensorValue(
229 return ([sparse_tensor.SparseTensor.from_value(indices0),
230 sparse_tensor.SparseTensor.from_value(indices1)],
231 sparse_tensor.SparseTensor.from_value(values))
275 return sparse_tensor.SparseTensorValue(
280 return sparse_tensor.SparseTensor.from_value(self._SparseTensorValue_5x6())
323 return sparse_tensor.SparseTensor(
329 return sparse_tensor.SparseTensor(
336 return sparse_tensor.SparseTensorValue(self._IND_2_5_6, self._VAL_2_5_6,
457 return sparse_tensor.SparseTensorValue(
462 return sparse_tensor.SparseTensor.from_value(self._SparseTensorValue_5x6())
468 return sparse_tensor.SparseTensor(
477 return sparse_tensor.SparseTensor(
504 sp_input = sparse_tensor.SparseTensorValue(
573 sp_input = sparse_tensor.SparseTensor(indices, values, shape)
647 sp_t = sparse_tensor.SparseTensor(self.ind, self.vals, self.dense_shape)
671 sp_t = sparse_tensor.SparseTensor(self.ind, self.vals, self.dense_shape)
712 self.assertTrue(isinstance(result_tensor, sparse_tensor.SparseTensor))
713 self.assertTrue(isinstance(input_sp_t, sparse_tensor.SparseTensor))
753 sp_t = sparse_tensor.SparseTensor(indices, vals, shape)
868 sp_zero = sparse_tensor.SparseTensor([[0]], [0], [7])
869 sp_one = sparse_tensor.SparseTensor([[0]], [1], [7])
876 sp_zero = sparse_tensor.SparseTensor([[0]], [0], [7])
877 sp_zero_2 = sparse_tensor.SparseTensor([[1]], [0], [7])
878 expected = sparse_tensor.SparseTensor([[0], [1]], [0, 0], [7])
912 sp_zero = sparse_tensor.SparseTensor([[0, 0]], [0], [1, 1])
913 sp_one = sparse_tensor.SparseTensor([[0]], [1], [2])
917 sp_zero = sparse_tensor.SparseTensor([[0]], [0], [1])
918 sp_one = sparse_tensor.SparseTensor([[0]], [1], [2])
