Searched defs:batch_to_space (Results 1 – 5 of 5) sorted by relevance
38 SimpleTensor<T> batch_to_space(const SimpleTensor<T> &src, const SimpleTensor<int32_t> &block_shape… in batch_to_space() function
74 FunctionType batch_to_space; in compute_target() local
36 def batch_to_space(*args, **kwargs): member in PythonOpImpl43 def batch_to_space(*args, **kwargs): member in CppOpImpl
80 def batch_to_space(*args, **kwargs): member in PythonOpImpl91 def batch_to_space(*args, **kwargs): member in CppOpImpl
4148 def batch_to_space(input, crops, block_size, name=None, block_shape=None): # pylint: disable=redef… function