Searched refs:input_shape_or_tensor (Results 1 – 1 of 1) sorted by relevance
388 def NHWCToNCHW_VECT_C(input_shape_or_tensor): argument405 is_tensor = isinstance(input_shape_or_tensor, ops.Tensor)407 input_shape_or_tensor.shape.as_list()408 if is_tensor else input_shape_or_tensor)417 t = array_ops.reshape(input_shape_or_tensor, temp_shape)423 def NCHW_VECT_CToNHWC(input_shape_or_tensor): argument439 is_tensor = isinstance(input_shape_or_tensor, ops.Tensor)441 input_shape_or_tensor.shape.as_list()442 if is_tensor else input_shape_or_tensor)449 t = array_ops.transpose(input_shape_or_tensor, permutation)