Searched refs:internal_outputs_tensor_map_ (Results 1 – 2 of 2) sorted by relevance
94 internal_outputs_tensor_map_ = graph.internal_outputs_tensor_map_; in KernelGraph()456 …rdered_map<AnfNodePtr, std::unordered_map<size_t, tensor::TensorPtr>> internal_outputs_tensor_map_; variable
1059 internal_outputs_tensor_map_[node][output_idx] = tensor; in AddInternalOutputTensor()1066 auto iter = internal_outputs_tensor_map_.find(node); in GetInternalOutputTensor()1067 if (iter == internal_outputs_tensor_map_.end()) { in GetInternalOutputTensor()