Searched refs:output_lengths (Results 1 – 4 of 4) sorted by relevance
1647 const std::vector<size_t> &output_lengths) { in SetInputAndOutputTypeLength() argument1653 if (output_lengths.size() != outputs_shape_.size()) { in SetInputAndOutputTypeLength()1654 MS_LOG(ERROR) << "Output_lengths: " << output_lengths.size() in SetInputAndOutputTypeLength()1659 outputs_type_lengths_ = output_lengths; in SetInputAndOutputTypeLength()1660 operator_cost()->SetInputAndOutputTypeLength(input_lengths, output_lengths); in SetInputAndOutputTypeLength()
77 const std::vector<size_t> &output_lengths);
36 const std::vector<size_t> &output_lengths) { in SetInputAndOutputTypeLength() argument38 outputs_type_lengths_ = output_lengths; in SetInputAndOutputTypeLength()
66 …putTypeLength(const std::vector<size_t> &input_lengths, const std::vector<size_t> &output_lengths);