Searched refs:input_shape_or_tensor (Results 1 – 1 of 1) sorted by relevance
286 def NHWCToNCHW_VECT_C(input_shape_or_tensor): argument303 is_tensor = isinstance(input_shape_or_tensor, ops.Tensor)305 input_shape_or_tensor.shape.as_list()306 if is_tensor else input_shape_or_tensor)315 t = array_ops.reshape(input_shape_or_tensor, temp_shape)321 def NCHW_VECT_CToNHWC(input_shape_or_tensor): argument337 is_tensor = isinstance(input_shape_or_tensor, ops.Tensor)339 input_shape_or_tensor.shape.as_list()340 if is_tensor else input_shape_or_tensor)347 t = array_ops.transpose(input_shape_or_tensor, permutation)