Searched refs:tensor_reserver_ (Results 1 – 5 of 5) sorted by relevance
191 tensor_reserver_.Add( in ReserveGraphTensors()195 tensor_reserver_.SetNext(max_id + 1); in ReserveGraphTensors()208 tensor_descriptors[value->id] = tensor_reserver_.Get(value->id).descriptor; in Compile()226 tensor_reserver_.Get(outputs[0]->id).descriptor; in Compile()267 tensor_reserver_.Get(inputs[j]->id).descriptor); in Compile()271 tensor_reserver_.Get(outputs[j]->id).descriptor); in Compile()279 auto global_id = tensor_reserver_.Add({t.first, t.second}); in Compile()373 const auto& t = tensor_reserver_.Get(tensor_id); in AllocateTensors()419 src_shapes.push_back(tensor_reserver_.Get(in_id).shape); in UpdateParams()422 dst_shapes.push_back(tensor_reserver_.Get(out_id).shape); in UpdateParams()[all …]
203 TensorReserver tensor_reserver_; variable
284 tensor_reserver_.Add( in ReserveGraphTensors()288 tensor_reserver_.SetNext(max_id + 1); in ReserveGraphTensors()297 tensor_descriptors[value->id] = tensor_reserver_.Get(value->id).descriptor; in ConvertOperations()318 tensor_reserver_.Get(outputs[0]->id).descriptor; in ConvertOperations()359 tensor_reserver_.Get(inputs[j]->id).descriptor); in ConvertOperations()363 tensor_reserver_.Get(outputs[j]->id).descriptor); in ConvertOperations()371 auto global_id = tensor_reserver_.Add({t.first, t.second}); in ConvertOperations()478 } else if (IsBufferBased(tensor_reserver_.Get(id).descriptor.storage_type)) { in GetTensorMemoryType()509 const auto& t = tensor_reserver_.Get(value_and_ref_value.first); in AllocateMemoryForVariableTensors()531 const auto& t = tensor_reserver_.Get(usage.first); in AllocateMemoryForBuffers()[all …]
218 TensorReserver tensor_reserver_; variable
978 auto tensors = inference.tensor_reserver_.GetTensorDescs(); in Encode()1043 inference->tensor_reserver_.Add(tensors); in Decode()