Home
last modified time | relevance | path

Searched refs:id_tensor (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_optimizer.py199 id_tensor = column.id_tensor(transformed_tensor)
203 id_tensor = transformed_tensor
205 [array_ops.shape(id_tensor.indices)[0]], dtypes.float32)
207 example_ids = array_ops.reshape(id_tensor.indices[:, 0], [-1])
209 flat_ids = array_ops.reshape(id_tensor.values, [-1])
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py1992 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [
1993 array_ops.shape(sparse_tensors.id_tensor)[0], -1
2011 id_tensor,
2428 sparse_ids = sparse_tensors.id_tensor
2487 sparse_tensors.id_tensor)
2543 sparse_ids = sparse_tensors.id_tensor
2629 sparse_tensors.id_tensor)
2954 feature_tensors.append(ids_and_weights.id_tensor)
3017 id_tensor = id_weight_pair.id_tensor
3023 sp_ids=id_tensor,
[all …]
Dfeature_column_v2.py2447 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [
2448 array_ops.shape(sparse_tensors.id_tensor)[0], -1
2457 id_tensor,
3051 sparse_ids = sparse_tensors.id_tensor
3160 sparse_tensors.id_tensor)
3186 sparse_tensors.id_tensor)
3324 sparse_ids = sparse_tensors.id_tensor
3370 sparse_tensors.id_tensor)
4040 feature_tensors.append(ids_and_weights.id_tensor)
4063 feature_tensors.append(ids_and_weights.id_tensor)
[all …]
Dfeature_column_test.py511 id_tensor_value = sess.run(id_weight_pair.id_tensor)
526 id_tensor_value = sess.run(id_weight_pair.id_tensor)
842 self.assertEqual(builder.get(hashed_sparse), id_weight_pair.id_tensor)
865 self.assertEqual(builder.get(hashed_sparse), id_weight_pair.id_tensor)
1090 id_tensor_eval = id_weight_pair.id_tensor.eval()
1120 id_tensor_eval = id_weight_pair.id_tensor.eval()
1189 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
1277 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
3567 id_weight_pair.id_tensor.eval())
3586 id_weight_pair.id_tensor.eval())
[all …]
Dfeature_column_v2_test.py617 id_tensor_value = sess.run(id_weight_pair.id_tensor)
635 id_tensor_value = sess.run(id_weight_pair.id_tensor)
1008 transformation_cache.get(hashed_sparse, None), id_weight_pair.id_tensor)
1020 transformation_cache.get(hashed_sparse, None), id_weight_pair.id_tensor)
1275 id_tensor_eval = self.evaluate(id_weight_pair.id_tensor)
1307 id_tensor_eval = self.evaluate(id_weight_pair.id_tensor)
1384 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
1495 id_tensor=ids_and_weights[0], weight_tensor=ids_and_weights[1])
4720 self.evaluate(id_weight_pair.id_tensor))
4745 self.evaluate(id_weight_pair.id_tensor))
[all …]
Dsequence_feature_column_test.py750 self, expected, self.evaluate(id_weight_pair.id_tensor))
789 self, expected, self.evaluate(id_weight_pair.id_tensor))
844 self, expected, self.evaluate(id_weight_pair.id_tensor))
868 result = id_weight_pair.id_tensor.eval(
911 self, expected, self.evaluate(id_weight_pair.id_tensor))
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column.py414 def id_tensor(self, input_tensor): member in _SparseColumn
436 input_tensor=self.id_tensor(input_tensor),
497 self.id_tensor(input_tensor), self.weight_tensor(input_tensor))
811 def id_tensor(self, input_tensor): member in _WeightedSparseColumn
832 input_tensor=self.id_tensor(input_tensor),
838 def _do_transform(self, id_tensor, weight_tensor): argument
844 return tuple([id_tensor, weight_tensor])
873 self.id_tensor(input_tensor), self.weight_tensor(input_tensor))
987 sparse_id_column = self.sparse_id_column.id_tensor(transformed_input_tensor)
1148 input_tensor=self.sparse_id_column.id_tensor(input_tensor),
[all …]
Dfeature_column_test.py337 id_tensor = _sparse_id_tensor(id_tensor_shape, vocab_size)
342 id_tensor, output_rank=output_rank)
370 id_tensor, weight_tensor = _sparse_id_tensor_with_weights(
374 (id_tensor, weight_tensor),
Dfeature_column_ops_test.py1554 id_tensor = sparse_tensor.SparseTensor(
1566 "id": id_tensor
/external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/
Dsequence_feature_column_test.py703 self, expected, id_weight_pair.id_tensor.eval(session=sess))
742 self, expected, id_weight_pair.id_tensor.eval(session=sess))
795 self, expected, id_weight_pair.id_tensor.eval(session=sess))
819 result = id_weight_pair.id_tensor.eval(
860 self, expected, id_weight_pair.id_tensor.eval(session=sess))