Searched refs:device_tensors_ (Results 1 – 2 of 2) sorted by relevance
174 std::vector<PersistentTensor> device_tensors_; variable
507 calib_ctx->device_tensors_.at(i).AccessTensor(ctx); in ExecuteCalibration()1026 cres->device_tensors_.resize(num_inputs); in AllocateCalibrationResources()1034 t.dtype(), t.shape(), &cres->device_tensors_.at(i), &device_tensor)); in AllocateCalibrationResources()