/external/tensorflow/tensorflow/python/tpu/ |
D | feature_column_test.py | 57 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/ |
D | sequence_feature_column.py | 123 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/ |
D | feature_column.py | 206 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 …]
|
D | feature_column_v2.py | 2782 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
|
D | feature_column_test.py | 437 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 …]
|
D | feature_column_v2_test.py | 2438 def _variable_shape(self): member in OldLinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
|
/external/tensorflow/tensorflow/contrib/layers/python/layers/ |
D | feature_column.py | 1021 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
|