Searched refs:input_shape_or_tensor (Results 1 – 1 of 1) sorted by relevance
354 def NHWCToNCHW_VECT_C(input_shape_or_tensor): argument371 is_tensor = isinstance(input_shape_or_tensor, ops.Tensor)373 input_shape_or_tensor.shape.as_list()374 if is_tensor else input_shape_or_tensor)383 t = array_ops.reshape(input_shape_or_tensor, temp_shape)389 def NCHW_VECT_CToNHWC(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)415 t = array_ops.transpose(input_shape_or_tensor, permutation)