Searched refs:flatten_dims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1964 @RegisterPForWithArgs("Conv2DBackpropInput", flatten_dims=[2], shape_dim=0)
1965 @RegisterPForWithArgs("AvgPoolGrad", flatten_dims=[1], shape_dim=0)
1966 @RegisterPForWithArgs("AvgPool3DGrad", flatten_dims=[1], shape_dim=0)
1967 def _convert_flatten_batch_shape_input(pfor_input, op_type, flatten_dims, argument
1970 inputs = _inputs_with_flattening(pfor_input, flatten_dims)
|