Searched defs:input_indices (Results 1 – 10 of 10) sorted by relevance
35 …void set_input_indices(const std::vector<size_t> &input_indices) { input_indices_ = input_indices;… in set_input_indices()37 const std::vector<size_t> &input_indices() const { return input_indices_; } in input_indices() function
38 void AddOutputs(const CNodePtr &cnode, const std::vector<size_t> &input_indices) { in AddOutputs()
103 …putToOutputRegistry::Register(const std::string &op_name, const std::vector<size_t> &input_indices, in Register()
132 std::vector<uint32_t> input_indices; in InitGraphInOutTensors() local216 void CoderGraph::SetInputIndices(const std::vector<uint32_t> &input_indices) { in SetInputIndices()244 std::vector<uint32_t> CoderGraph::input_indices() const { return this->input_indices_; } in input_indices() function in mindspore::lite::micro::CoderGraph
251 std::vector<uint32_t> input_indices; in CreateOpCoders() local
30 def construct(self, input_params, input_indices, offset): argument
34 void OperatorCoder::set_input_tensor_indices(const std::vector<uint32_t> &input_indices) { in set_input_tensor_indices()
92 OpCoderBuilder &OpCoderBuilder::input_indices(const std::vector<uint32_t> &indices) { in input_indices() function in mindspore::lite::micro::OpCoderBuilder
103 auto input_indices = reinterpret_cast<T *>(inputs[1]->addr); in LaunchKernel() local
550 def construct(self, input_indices, input_values, field_ids): argument