Searched refs:external_mutable_tensors (Results 1 – 5 of 5) sorted by relevance
101 create_info->external_mutable_tensors.find(id) != in GetTensorType()102 create_info->external_mutable_tensors.end())) { in GetTensorType()272 for (const auto& external_tensor : create_info.external_mutable_tensors) { in InitFromGpuModel()369 for (const auto& external_tensor : create_info->external_mutable_tensors) { in RestoreDeserialized()
95 absl::flat_hash_map<ValueId, TensorDescriptor> external_mutable_tensors; member
205 auto it_mutable_external = create_info.external_mutable_tensors.find(t->id); in ReserveGraphTensors()216 if (it_mutable_external != create_info.external_mutable_tensors.end()) { in ReserveGraphTensors()
367 create_info.external_mutable_tensors[value->id] =372 create_info.external_mutable_tensors[value->id] =407 TensorDescriptor descriptor_with_shape = create_info.external_mutable_tensors[input];439 TensorDescriptor descriptor_with_shape = create_info.external_mutable_tensors[output];
164 for (const auto& external_tensor : create_info.external_mutable_tensors) { in InitFromGraph()239 for (const auto& external_tensor : create_info->external_mutable_tensors) { in RestoreDeserialized()