Searched defs:reshape_with_axis (Results 1 – 3 of 3) sorted by relevance
28 def __init__(self, cell, time_axis, reshape_with_axis=None): argument
104 def __init__(self, layer, time_axis, reshape_with_axis=None): argument