Searched refs:workspace_size_ (Results 1 – 1 of 1) sorted by relevance
131 int64 workspace_size_; member in tensorflow::tensorrt::TRTEngineOp192 context->GetAttr("workspace_size_bytes", &workspace_size_)); in TRTEngineOp()624 segment_graph_, precision_mode_, batch_size, workspace_size_, in GetEngine()693 const int64 workspace_size_bytes = workspace_size_; in AllocateCalibrationResources()