Searched defs:original_output_shape (Results 1 – 2 of 2) sorted by relevance
303 std::vector<int64_t> original_output_shape{act.sizes().vec()}; // 2D in apply_impl() local
1618 const auto& original_output_shape = output.shape(); in CalculateOutputSize() local1647 const auto& original_output_shape = output.shape(); in CalculateOutputTensorSize() local