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> row_shape; in AsGraphDefInternal() local173 const PartialTensorShape& row_shape = in GetNextInternal() local
102 def dense_to_sparse_batch(batch_size, row_shape): argument300 def __init__(self, input_dataset, batch_size, row_shape): argument
2493 TensorShape dense_shape, row_shape; in ParseSequenceDenseFeatures() local