Searched defs:values_shape (Results 1 – 6 of 6) sorted by relevance
33 def _has_valid_dims(weights_shape, values_shape): argument47 weights_rank, weights_shape, values_rank, values_shape): argument
522 TensorShape values_shape({0}); in Compute() local632 TensorShape values_shape({total_num_features}); in Compute() local
1006 TensorShape values_shape; in FastParseExample() local1081 TensorShape values_shape; in FastParseExample() local1170 TensorShape values_shape; in FastParseSingleExample() local1367 TensorShape values_shape; in FastParseSingleExample() local
86 TensorShape values_shape; member
379 const TensorShape& values_shape = sparse_tensor_batch_shapes.values_shape; in BatchExampleProtoToTensors() local
1287 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor()