Searched refs:flatten_dims (Results 1 – 1 of 1) sorted by relevance
1566 @RegisterPForWithArgs("Conv2DBackpropInput", flatten_dims=[2], shape_dim=0)1567 @RegisterPForWithArgs("AvgPoolGrad", flatten_dims=[1], shape_dim=0)1568 def _convert_flatten_batch_shape_input(pfor_input, op_type, flatten_dims, argument1571 inputs = _inputs_with_flattening(pfor_input, flatten_dims)