Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:_sparse_tensor

36         self._sparse_tensor([['batch1-FC1-F1'],
38 self._sparse_tensor([['batch1-FC2-F1'],
41 expected_out = self._sparse_tensor([['batch1-FC1-F1_X_batch1-FC2-F1'], [
58 expected_out = self._sparse_tensor([[
71 self._sparse_tensor([[11], [333, 55555]]),
72 self._sparse_tensor([['batch1-FC2-F1'],
75 expected_out = self._sparse_tensor([['11_X_batch1-FC2-F1'], [
90 expected_out = self._sparse_tensor([[
103 self._sparse_tensor([['batch1-FC1-F1'],
109 expected_out = self._sparse_tensor(
120 self._sparse_tensor([[11], [333, 5555]]),
125 expected_out = self._sparse_tensor(
136 self._sparse_tensor(
138 self._sparse_tensor(
140 self._sparse_tensor(
143 expected_out = self._sparse_tensor([[
178 self._sparse_tensor(
180 self._sparse_tensor([['batch1-FC2-F1']]),
181 self._sparse_tensor([['batch1-FC3-F1', 'batch1-FC3-F2']])
183 expected_out = self._sparse_tensor([[
207 self._sparse_tensor(col1), self._sparse_tensor(col2),
208 self._sparse_tensor(col3)
222 expected_out = self._sparse_tensor(col_out)
232 self._sparse_tensor([['batch1-FC1-F1', 'batch1-FC1-F2']]),
233 self._sparse_tensor([], 1),
234 self._sparse_tensor([['batch1-FC3-F1', 'batch1-FC3-F2']])
245 self._sparse_tensor([['batch1-FC1-F1', 'batch1-FC1-F2']], 2),
246 self._sparse_tensor([['batch1-FC2-F1'], ['batch2-FC2-F1']], 2),
247 self._sparse_tensor([['batch1-FC3-F1', 'batch1-FC3-F2']], 2)
249 expected_out = self._sparse_tensor([[
264 self._sparse_tensor([]), self._sparse_tensor([]),
265 self._sparse_tensor([])
273 self._sparse_tensor([['batch1-FC1-F1']]),
274 self._sparse_tensor([['batch1-FC2-F1']]),
275 self._sparse_tensor([['batch1-FC3-F1']])
278 expected_out = self._sparse_tensor([[1971693436396284976]])
285 self._sparse_tensor([['batch1-FC1-F1']]),
286 self._sparse_tensor([['batch1-FC2-F1']]),
287 self._sparse_tensor([['batch1-FC3-F1']])
291 expected_out = self._sparse_tensor([[4847552627144134031]])
299 self._sparse_tensor([['batch1-FC1-F1']]),
300 self._sparse_tensor([['batch1-FC2-F1']]),
301 self._sparse_tensor([['batch1-FC3-F1']])
305 expected_out = self._sparse_tensor([[83]])
312 self._sparse_tensor([['batch1-FC1-F1']]),
313 self._sparse_tensor([['batch1-FC2-F1']]),
314 self._sparse_tensor([['batch1-FC3-F1']])
319 expected_out = self._sparse_tensor([[31]])
342 self._sparse_tensor(
344 self._sparse_tensor([['batch1-FC2-F1']]),
345 self._sparse_tensor([['batch1-FC3-F1', 'batch1-FC3-F2']])
367 def _sparse_tensor(self, data, batch_size=-1):