Searched defs:row_shape (Results 1 – 3 of 3) sorted by relevance
50 PartialTensorShape row_shape; in MakeDataset() local80 const PartialTensorShape& row_shape, const DatasetBase* input) in Dataset()142 std::vector<int64_t> row_shape; in AsGraphDefInternal() local173 const PartialTensorShape& row_shape = in GetNextInternal() local
99 def dense_to_sparse_batch(batch_size, row_shape): argument304 def __init__(self, input_dataset, batch_size, row_shape): argument
2510 TensorShape dense_shape, row_shape; in ParseSequenceDenseFeatures() local