Home
last modified time | relevance | path

Searched refs:_variable_shape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_test.py57 self.assertEqual((embedding_dimension,), embedding_column._variable_shape)
76 self.assertEqual((embedding_dimension,), embedding_column._variable_shape)
173 self.assertEqual((embedding_dimension,), embedding_column_a._variable_shape)
174 self.assertEqual((embedding_dimension,), embedding_column_b._variable_shape)
211 self.assertEqual((embedding_dimension,), embedding_column_a._variable_shape)
212 self.assertEqual((embedding_dimension,), embedding_column_b._variable_shape)
/external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/
Dsequence_feature_column.py123 num_elements = column._variable_shape.num_elements()
486 def _variable_shape(self): member in _SequenceNumericColumn
500 [array_ops.shape(dense_tensor)[:1], [-1], self._variable_shape],
508 num_elements = (self._variable_shape.num_elements()
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py206 num_elements = column._variable_shape.num_elements() # pylint: disable=protected-access
552 …num_elements = self._feature_column._variable_shape.num_elements() # pylint: disable=protected-ac…
1825 def _variable_shape(self): member in _DenseColumn
1893 num_elements = column._variable_shape.num_elements() # pylint: disable=protected-access
2298 def _variable_shape(self): member in _NumericColumn
2343 def _variable_shape(self): member in _BucketizedColumn
2415 def _variable_shape(self): member in _EmbeddingColumn
2526 def _variable_shape(self): member in _SharedEmbeddingColumn
3025 vocab_size=int(self._variable_shape[-1]))
3042 depth=self._variable_shape[-1],
[all …]
Dfeature_column_v2.py2782 def _variable_shape(self): member in NumericColumn
2886 def _variable_shape(self): member in BucketizedColumn
3033 def _variable_shape(self): member in EmbeddingColumn
3312 def _variable_shape(self): member in SharedEmbeddingColumn
4191 vocab_size=int(self._variable_shape[-1]))
4208 depth=self._variable_shape[-1],
4262 def _variable_shape(self): member in IndicatorColumn
Dfeature_column_test.py437 self.assertAllEqual((2, 3), b._variable_shape)
551 self.assertAllEqual(a_bucketized_copy._variable_shape, (2, 3))
1447 def _variable_shape(self): member in LinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
2128 def _variable_shape(self): member in _LinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
4575 self.assertEqual(indicator_a._variable_shape, [1, 4])
4582 self.assertEqual(indicator_b._variable_shape, [1, 100])
4641 self.assertEqual(column._variable_shape, [1, 4])
4800 (embedding_dimension,), embedding_column._variable_shape)
4829 (embedding_dimension,), embedding_column._variable_shape)
4863 (embedding_dimension,), embedding_column._variable_shape)
[all …]
Dfeature_column_v2_test.py2438 def _variable_shape(self): member in OldLinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column.py1021 def _variable_shape(self): member in _OneHotColumn
1170 def _variable_shape(self): member in _EmbeddingColumn
1541 def _variable_shape(self): member in _ScatteredEmbeddingColumn
1891 def _variable_shape(self): member in _RealValuedColumn
2185 def _variable_shape(self): member in _BucketizedColumn