Searched defs:tensor_id (Results 1 – 7 of 7) sorted by relevance
35 auto tensor_id = node->inputIndex.at(i); in Run() local44 auto tensor_id = node->outputIndex.at(i); in Run() local
29 AnfNodePtr AnfImporterFromMetaGraphT::GetNode(int tensor_id) { in GetNode()37 void AnfImporterFromMetaGraphT::AddNode(int tensor_id, AnfNodePtr node) { nodes_[tensor_id] = std::… in AddNode()
44 uint64 tensor_id = 1; // tensor id field
59 void SetTensorId(uint64_t tensor_id) { tensor_id_ = tensor_id; } in SetTensorId()
337 def _update_mem_change(self, size, start, end, tensor_id): argument
283 auto tensor_id = tensor_json[kTensorId]; in UpdateTensorsOffset() local1174 for (auto tensor_id : contiguous_list) { in Assign() local
1768 …[&forward_op_tensor_id](const std::string &tensor_id) { forward_op_tensor_id.emplace(tensor_id); }… in SaveForwardTensorInfoInBpropGraph()