Searched refs:_batch_flatten_shape_func (Results 1 – 3 of 3) sorted by relevance
934 def _batch_flatten_shape_func(data_shape): function949 return [_batch_flatten_shape_func(inputs[0])]
913 def _batch_flatten_shape_func(data_shape): function927 return [_batch_flatten_shape_func(inputs[0])]