Searched defs:row_shape (Results 1 – 4 of 4) sorted by relevance
52 PartialTensorShape row_shape; in MakeDataset() local82 const PartialTensorShape& row_shape, const DatasetBase* input) in Dataset()134 std::vector<int64> row_shape; in AsGraphDefInternal() local165 const PartialTensorShape& row_shape = in GetNextInternal() local
29 def dense_to_sparse_batch(batch_size, row_shape): argument
156 def dense_to_sparse_batch(batch_size, row_shape): argument451 def __init__(self, input_dataset, batch_size, row_shape): argument
2198 TensorShape dense_shape, row_shape; in FastParseSequenceExample() local