Searched refs:AddThunkToThunkSequence (Results 1 – 3 of 3) sorted by relevance
34 virtual void AddThunkToThunkSequence(std::unique_ptr<Thunk> thunk) = 0;51 void AddThunkToThunkSequence(std::unique_ptr<Thunk> thunk) { in AddThunkToThunkSequence() function52 return context_->AddThunkToThunkSequence(std::move(thunk)); in AddThunkToThunkSequence()
802 AddThunkToThunkSequence(std::move(thunk)); in HandleConditional()807 AddThunkToThunkSequence( in HandleConvolution()1217 AddThunkToThunkSequence(absl::make_unique<ConvolutionThunk>( in EmitConvolutionThunkFromMlir()1287 AddThunkToThunkSequence(absl::make_unique<SequentialThunk>( in EmitGemmThunkFromMlir()1293 AddThunkToThunkSequence(absl::make_unique<GemmThunk>( in EmitGemmThunkFromMlir()1378 AddThunkToThunkSequence( in EmitBatchNormThunkFromMlir()1429 AddThunkToThunkSequence(absl::make_unique<CudnnBatchNormBackwardThunk>( in EmitBatchNormThunkFromMlir()1468 AddThunkToThunkSequence(absl::make_unique< in EmitBatchNormThunkFromMlir()1524 AddThunkToThunkSequence(std::move(thunks[0])); in EmitCholeskyThunkFromMlir()1526 AddThunkToThunkSequence(absl::make_unique<SequentialThunk>( in EmitCholeskyThunkFromMlir()[all …]
245 void AddThunkToThunkSequence(std::unique_ptr<Thunk> thunk) override { in AddThunkToThunkSequence() function