Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_test.py57 self.assertEqual((embedding_dimension,), embedding_column._variable_shape)
83 self.assertEqual((embedding_dimension,), embedding_column._variable_shape)
101 self.assertEqual((embedding_dimension,), embedding_column._variable_shape)
198 self.assertEqual((embedding_dimension,), embedding_column_a._variable_shape)
199 self.assertEqual((embedding_dimension,), embedding_column_b._variable_shape)
236 self.assertEqual((embedding_dimension,), embedding_column_a._variable_shape)
237 self.assertEqual((embedding_dimension,), embedding_column_b._variable_shape)
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py207 num_elements = column._variable_shape.num_elements() # pylint: disable=protected-access
558 …num_elements = self._feature_column._variable_shape.num_elements() # pylint: disable=protected-ac…
1883 def _variable_shape(self): member in _DenseColumn
1951 num_elements = column._variable_shape.num_elements() # pylint: disable=protected-access
2356 def _variable_shape(self): member in _NumericColumn
2401 def _variable_shape(self): member in _BucketizedColumn
2496 def _variable_shape(self): member in _EmbeddingColumn
2614 def _variable_shape(self): member in _SharedEmbeddingColumn
3109 vocab_size=int(self._variable_shape[-1]))
3126 depth=self._variable_shape[-1],
[all …]
Dfeature_column_v2.py2898 def _variable_shape(self): member in NumericColumn
3004 def _variable_shape(self): member in BucketizedColumn
3176 def _variable_shape(self): member in EmbeddingColumn
3487 def _variable_shape(self): member in SharedEmbeddingColumn
4411 self._variable_shape[-1])
4435 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
4617 self.assertEqual(indicator_a._variable_shape, [1, 4])
4624 self.assertEqual(indicator_b._variable_shape, [1, 100])
4683 self.assertEqual(column._variable_shape, [1, 4])
4842 (embedding_dimension,), embedding_column._variable_shape)
4871 (embedding_dimension,), embedding_column._variable_shape)
4905 (embedding_dimension,), embedding_column._variable_shape)
[all …]
Dfeature_column_v2_test.py2451 def _variable_shape(self): member in OldLinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn