Searched refs:tensor_release_map (Results 1 – 2 of 2) sorted by relevance
176 std::map<int, int>* tensor_release_map; member308 auto it = shared_info->tensor_release_map->find(tensor_index); in ShouldPersistTensorflowTensor()309 return it != shared_info->tensor_release_map->end() && in ShouldPersistTensorflowTensor()450 op_data_->shared_info.tensor_release_map = in Init()505 if (op_data_->shared_info.tensor_release_map->find(tensor_index) != in Init()506 op_data_->shared_info.tensor_release_map->end()) { in Init()508 std::max(op_data_->shared_info.tensor_release_map->at(tensor_index), in Init()511 (*op_data_->shared_info.tensor_release_map)[tensor_index] = node_id; in Init()806 return *(op_data_->shared_info.tensor_release_map); in GetTensorReleaseMap()
145 const auto& tensor_release_map = GetTensorReleaseMap(delegate_kernel); in TEST_F() local148 tensor_release_map, in TEST_F()