Searched defs:col_params (Results 1 – 12 of 12) sorted by relevance
267 const CollectiveParams* col_params, in ExecuteAsync()393 const CollectiveParams& col_params, in CreateCollective()434 const CollectiveParams& col_params) { in CheckDependencies()447 const CollectiveParams& col_params) { in WaitForDependencies()456 const CollectiveParams& col_params) { in UnblockDependencies()
52 Status InitializeCollectiveParams(CollectiveParams* col_params) override { in InitializeCollectiveParams()
109 Status GenerateSubdivsInCollectiveParams(CollectiveParams* col_params) { in GenerateSubdivsInCollectiveParams()165 Status RingAlg::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
54 string SubdivPermDebugString(const CollectiveParams& col_params) { in SubdivPermDebugString()
45 Status RingGatherer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
48 Status RingReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
78 CollectiveParams* col_params) { in InitializeCollectiveParams()
29 Status NcclBase::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
522 auto col_params = new CollectiveParams(); in ComputeAsync() local657 auto col_params = new CollectiveParams(); in ComputeAsync() local785 auto col_params = new CollectiveParams(); in ComputeAsync() local911 auto col_params = new CollectiveParams(); in ComputeAsync() local
301 const CollectiveParams* col_params, in ExecuteAsync()331 virtual void WaitForDependencies(const CollectiveParams& col_params) {} in WaitForDependencies()335 virtual void UnblockDependencies(const CollectiveParams& col_params) {} in UnblockDependencies()370 const CollectiveParams* col_params; // Not owned member
172 OpKernelContext::Params* op_params, const CollectiveParams* col_params, in CollectiveContext()
72 const CollectiveParams* col_params = col_ctx->col_params; in Enqueue() local