Searched refs:workspace_buffer (Results 1 – 3 of 3) sorted by relevance
37 BufferAllocation::Slice workspace_buffer, in CholeskyThunk() argument44 workspace_buffer_(workspace_buffer), in CholeskyThunk()
46 BufferAllocation::Slice workspace_buffer,
1501 TF_ASSIGN_OR_RETURN(auto workspace_buffer, in EmitCholeskyThunkFromMlir()1519 input.thunk_info, options, a_buffer, workspace_buffer, info_buffer, in EmitCholeskyThunkFromMlir()