Searched refs:workspace_size_ (Results 1 – 2 of 2) sorted by relevance
194 int64 workspace_size_; member in tensorflow::tensorrt::TRTEngineOp334 context->GetAttr("workspace_size_bytes", &workspace_size_)); in TRTEngineOp()844 segment_graph_def_, precision_mode_, batch_size, workspace_size_, in BuildEngine()1080 cres->calibrator_->getBatchSize(), this->workspace_size_, in AllocateCalibrationResources()
3371 : index_(index), algo_(algo), workspace_size_(workspace_size) {} in DoBlasInternalImpl()3375 size_t workspace_size() const override { return workspace_size_; } in DoBlasInternalImpl()3388 size_t workspace_size_; in DoBlasInternalImpl() member in stream_executor::gpu::__anone9de0a230311::CUDABlasLtMatmulAlgorithm