Searched refs:input_shape_or_tensor (Results 1 – 1 of 1) sorted by relevance
390 def NHWCToNCHW_VECT_C(input_shape_or_tensor): argument407 is_tensor = isinstance(input_shape_or_tensor, ops.Tensor)409 input_shape_or_tensor.shape.as_list()410 if is_tensor else input_shape_or_tensor)419 t = array_ops.reshape(input_shape_or_tensor, temp_shape)425 def NCHW_VECT_CToNHWC(input_shape_or_tensor): argument441 is_tensor = isinstance(input_shape_or_tensor, ops.Tensor)443 input_shape_or_tensor.shape.as_list()444 if is_tensor else input_shape_or_tensor)451 t = array_ops.transpose(input_shape_or_tensor, permutation)