Searched refs:original_shape (Results 1 – 3 of 3) sorted by relevance
64 def get_reduce_axis(original_shape, broadcast_shape): argument66 if len(original_shape) > len(broadcast_shape):69 tmp_shape = [1] * (len(broadcast_shape) - len(original_shape)) + original_shape
107 auto original_shape = std::make_shared<std::vector<size_t>>(); in InitOptimInputsShape() local108 MS_EXCEPTION_IF_NULL(original_shape); in InitOptimInputsShape()110 original_inputs_shape->push_back(original_shape); in InitOptimInputsShape()114 original_shape->push_back(values[val_idx++]); in InitOptimInputsShape()
1074 original_shape = a.shape1090 a = a.reshape(original_shape)