Searched defs:status_tensor (Results 1 – 7 of 7) sorted by relevance
47 TensorPtr status_tensor; member196 const TensorPtr &status_tensor() const { return status_tensor_; } in status_tensor() function202 void set_status_tensor(const TensorPtr status_tensor) { status_tensor_ = status_tensor; } in set_status_tensor()
217 auto status_tensor = std::make_shared<Tensor>(kNumberTypeInt, key_shape); in Export() local
241 def _store_warm_up_ptr_by_tensor_list(param_key, key_tensor, value_tensor, status_tensor): argument
222 required TensorProto status_tensor = 5; field
240 auto status_tensor = map_tensor->status_tensor(); in CopyMapTensorDataToDevice() local
790 auto &status_tensor = export_data.status_tensor; in ConvertMapParameterToMapTensorProto() local
916 auto status_tensor = GenerateTensorPtrFromTensorProto(status_tensor_proto); in BuildMapParameterFromMapTensorProto() local