Searched refs:op_node_ (Results 1 – 7 of 7) sorted by relevance
323 op_node_.outputs.push_back(hexagon_nn_output()); in AddOutput()324 op_node_.outputs.back().elementsize = element_size; in AddOutput()325 op_node_.outputs.back().rank = 4; in AddOutput()329 auto& max_sizes = op_node_.outputs.back().max_sizes; in AddOutput()338 return TensorID(GetID(), op_node_.outputs.size() - 1); in AddOutput()343 op_node_.outputs.push_back(hexagon_nn_output()); in AddOutput()344 op_node_.outputs.back().elementsize = elementsize; in AddOutput()345 op_node_.outputs.back().rank = rank; in AddOutput()346 auto& max_sizes = op_node_.outputs.back().max_sizes; in AddOutput()353 return TensorID(GetID(), op_node_.outputs.size() - 1); in AddOutput()[all …]
69 op_node_.op_type = hexagon_op_type; in OpBuilder()80 void SetOpType(int op_type) { op_node_.op_type = op_type; } in SetOpType()83 void SetNodeId(int node_id) { op_node_.node_id = node_id; } in SetNodeId()87 op_node_.tflite_node_index = node_index; in SetTFLiteNodeId()91 void SetConstNode() { op_node_.op_type = OP_Const; } in SetConstNode()95 op_node_.padding_type = padding_type; in SetPaddingType()102 bool IsConstNode() const { return op_node_.op_type == OP_Const; } in IsConstNode()123 int GetTFLiteNodeID() const { return op_node_.tflite_node_index; } in GetTFLiteNodeID()126 int GetOpType() const { return op_node_.op_type; } in GetOpType()129 int GetID() const { return op_node_.node_id; } in GetID()[all …]
58 if (op_node_.op_type == OP_QuantizedAdd_8p8to8 && output_max_ != 0) { in PopulateSubGraph()64 if (op_node_.op_type == OP_QuantizedMul_8x8to32) { in PopulateSubGraph()
38 if (op_node_.op_type == OP_QuantizedReluX_8) { in PopulateSubGraph()
167 if (op_node_.op_type == OP_Supernode_8x8p32to8) { in InitializeWeightsNodes()201 } else if (op_node_.op_type == OP_DepthwiseSupernode_8x8p32to8) { in InitializeWeightsNodes()
66 if (op_node_.op_type == OP_QuantizedMaxPool_8) { in PopulateSubGraph()
147 if (op_node_.op_type == OP_Supernode_8x8p32to8) { in PopulateSubGraph()154 } else if (op_node_.op_type == OP_DepthwiseSupernode_8x8p32to8) { in PopulateSubGraph()