Searched refs:FlatCellStep (Results 1 – 1 of 1) sorted by relevance
71 def FlatCellStep(*flat_inputs): function105 pure_flat_cell_step = _ToPureFunction(FlatCellStep)138 self._theta = FlatCellStep.captured_inputs